python - 如何打开当前目录下的所有.txt和.log文件，搜索并打印搜索到的文件

Question

我正在尝试在当前目录的所有文本和日志文件中搜索字符串。如果找到匹配项，则打印找到匹配项的文本或日志文件。这可能吗？我该如何操作下面的代码来完成这项任务？

  fiLe = open(logfile, "r")
  userString = raw_input("Enter a string name to search: ")
  for line in fiLe.readlines():
      if userString in line:
         print line

Answer 1

像这样:

import os
directory = os.path.join("c:\\","path")
for root,dirs,files in os.walk(directory):
    for file in files:
       if file.endswith(".log") or file.endswith(".txt"):
           f=open(file, 'r')
           for line in f:
              if userstring in line:
                 print "file: " + os.path.join(root,file)             
                 break
           f.close()