python - 使用 os.walk 在文件中搜索字符串

Question

我编写了一个 Python 脚本，用于从目录中的每个文件中找出包含特定脚本的行。如果我让这个脚本在包含这些文件的目录中运行，它就可以正常工作。

#!/usr/bin/env python
import os

def searchthis(location, searchterm):
    for fname in os.listdir(location):
        fullpath = os.path.join(location, fname)
        for line in file(fullpath):
            if searchterm in line:
                print line



searchthis(os.getcwd(), "mystring")

有什么方法可以使用 os.walk 并在所有目录和子目录中存在的每个文件中递归搜索。

Answer 1

您可以像这样使用一个简单的迭代器:

def all_files(dir):
    for root, dirs, files in os.walk(os.path.abspath(dir)):
        for f in files:
            yield os.path.join(root, f)

例如:

 for path in all_files(os.getcwd()):
     with open(path) as f:
         for n, line in enumerate(f, 1):
             if term in line:
                 print path, n