《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("查找完成")