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