Py 在文件夹和子文件夹中搜索文件

Question

我正在尝试在目录树中查找文件列表。本质上，我提供了一个文本文件，其中包含我想要搜索的所有术语 (~500)，并让它在目录和子目录中查找它们。但是，我有问题 - 我相信 - 代码采取和过早结束的步骤而没有在所有文件夹中搜索。

我使用的代码是（pattern 是文本文件的名称）：

import os

def locateA(pattern, root):
    file  = open(pattern, 'r')
    for path, dirs, files in os.walk(root):
        for word in files:
            for line in file:
                if line.strip() in word:
                    print os.path.join(path, word), line.strip()

关于我错在哪里的任何想法？

Answer 1

问题的全部或部分可能是您只能遍历文件一次，除非您使用file.seek() 重置文件中的当前位置。

确保在尝试再次循环之前返回文件的开头：

import os

def locateA(pattern, root):
    file  = open(pattern, 'r')
    for path, dirs, files in os.walk(root):
        for word in files:
            file.seek(0)             # this line is new
            for line in file:
                if line.strip() in word:
                    print os.path.join(path, word), line.strip()

Answer 2

for line in file 第一次使用file 中的行，然后每次都为空。

试试这个，它可以解决这个问题和其他一些问题：

import os

def locateA(pattern, root):
    patterns = open(pattern, 'r').readlines() # patterns is now an array, no need to reread every time.
    for path, dirs, files in os.walk(root):
        for filename in files:
            for pattern in patterns:
                if pattern.strip() in filename:
                    print os.path.join(path, filename), pattern.strip()