Python實現對文件夾內文本文件遞歸查找

nf456 9年前發布 | 21K 次閱讀 Python Python開發

平臺: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 ')

程序雖小,但很實用,運行結果如下:

 本文由用戶 nf456 自行上傳分享,僅供網友學習交流。所有權歸原作者,若您的權利被侵害,請聯系管理員。
 轉載本站原創文章,請注明出處,并保留原始鏈接、圖片水印。
 本站是一個以用戶分享為主的開源技術平臺,歡迎各類分享!