Python實現對檔案夾內文字檔遞迴尋找

標籤：pen   ble   .cpp   std   sts   ror   odi   open   com   

經常有這樣的需求：在一個文字檔裡尋找特定字串，這很好實現，用任何文本查看工具幾乎都可以做到。而有的時候，想尋找一個檔案夾下的所有文字檔（特定尾碼），我就遇到了這樣的問題：想找到Blender的原始碼中關於SPH的實現代碼。於是寫了下面的簡單程式：

#!/usr/bin/env python3import osdef 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)                        fh.close()                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 ‘)

程式雖小，但很實用，運行結果如下：

 

Python實現對檔案夾內文字檔遞迴尋找