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

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

复制代码
#!/usr/bin/env python3

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)
                        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 ')
复制代码

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

 

posted @ 2017-06-03 17:51  赵哲丽  阅读(1579)  评论(0编辑  收藏  举报
友情链接:回力旗舰店 | 中老年女装 | 新航道英语培训 | SAT培训机构 | 十月英文