《Python编程快速上手》第8.9.3实践练习

#!python3
# -*- coding:utf-8 -*-

# 8.9.3
#打开指定路径中所有.txt文件
#用户输入正则，将匹配行输出，输出文件名

import re,os

pat=input("输入要处理的文件夹绝对路径：")
lis_dir=os.listdir(pat)
lis_txt=[]
for x in lis_dir:
    #取出.txt结尾的文件名,加进列表
    if re.search(r'\.txt$',x):
        lis_txt.append(x)

text=input("输入你的自定义正则表达式：")
regCom=re.compile(text)
#regCom=re.compile('.*name.*')

#循环处理文件
for fi in lis_txt:
    fi=os.path.join(pat,fi)
    fi_open=open(fi)
    for fi_line in fi_open.readlines():
        fi_reg=regCom.search(fi_line)
        if fi_reg:
            print(fi_reg.group())  
    fi_open.close()
    
print("查找完成")