如何打开当前目录中的所有.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()