8.9.3 正则表达查找 程序答案

编写一个程序，打开文件夹中所有的.txt文件，查找匹配用户提供的正则表达式的所有行。结果应该打印到屏幕上

import re
import os
import shelve
import sys

cwd=os.getcwd()
txtDirList=[]
regex1=re.compile(r'\.txt$')
for x in os.listdir(cwd):
    if regex1.search(x)!=None:
        txtDirList.append(x)
print(txtDirList)
print(sys.argv[1])
regex2=re.compile(sys.argv[1])
txtLineList=[]
for x in txtDirList:
    with open(x) as txtFile:
        txtLineList=txtFile.readlines()
        for y in txtLineList:
            if regex2.search(y)!=None:
                print(y)