Python实现对文件夹内文本文件递归查找

nf456 10年前

平台:Win7 64 bit,IDLE Python 3.4.0

经常有这样的需求:在一个文本文件里查找特定字符串,这很好实现,用任何文本查看工具几乎都可以做到。而有的时候,想查找一个文件夹下的所有文本文件(特定后缀),我就遇到了这样的问题:想找到Blender的源代码中关于SPH的实现代码。于是写了下面的简单程序:

import os  def Search(rootDir, suffixes, arg):         for lists in os.listdir(rootDir):                 path = os.path.join(rootDir, lists)          if os.path.isfile(path):              if path.endswith(suffixes):                  try:                      with open(path, encoding='utf_8') as fh:                          lineNum = 0                          for line in fh:                              lineNum += 1                              if arg in line:                                  print(lineNum, ':', path, '\n', line)                  except:                      print('error: ', path, '\n')          if os.path.isdir(path):              Search(path, suffixes, arg)  Search(r'D:\blender-2.70', ('.c','.cpp','.h','.hpp'), 'SPH ')

程序虽小,但很实用,运行结果如下: