Python文本在不同文件夹中的.txt文件中搜索并打印文件和文件夹的名称

Question

我正在用 Python 编写一个脚本，用于在选定文件夹中的一堆 .txt 文件中搜索选定的术语（单词/单词、句子），并打印出包含选定项的 .txt 文件的名称学期。目前使用os 模块工作得很好：

import os

dirname = '/Users/User/Documents/test/reports'

search_terms = ['Pressure']
search_terms = [x.lower() for x in search_terms]

for f in os.listdir(dirname):
    with open(os.path.join(dirname,f), "r", encoding="latin-1") as infile:
        text =  infile.read()

    if all(term in text for term in search_terms):
        print (f)

但我想为脚本制作扩展名：不仅可以在一个文件夹（目录名）中搜索，还可以在两个（例如，目录名1，目录名2）中搜索，其中也包含 .txt 文件。此外，我不仅想打印搜索到的报告的名称，还想打印它所在目录的名称（dirname）。是否可以使用os 模块来做到这一点，或者会有其他一些方法来做到这一点？

Answer 1

您可以像这样遍历目录名：

import os

dirnames = ['/Users/User/Documents/test/reports','/Users/User/Documents/test/reports2']

search_terms = ['Pressure']
search_terms = [x.lower() for x in search_terms]
for dir_name in dirnames:
    for f in os.listdir(dir_name):
        with open(os.path.join(dir_name, f), "r", encoding="latin-1") as infile:
            text = infile.read()

        if all(term in text for term in search_terms):
            print("{} in {} directory".format(f, dir_name))