python - 在 Python 中查找包含所需字符串的目录中的文件

Question

我正在尝试在目录中包含的文件中查找字符串。我有一个像 banana 这样的字符串我知道存在于一些文件中。

import os
import sys

user_input = input("What is the name of you directory?")
directory = os.listdir(user_input)
searchString = input("What word are you trying to find?")

for fname in directory: # change directory as needed
    if searchString in fname:
        f = open(fname,'r')
        print('found string in file %s') %fname
    else:
        print('string not found')

程序运行时只输出 string not found对于每个文件。有三个文件包含单词 banana ，因此程序无法正常工作。为什么不在文件中找到字符串？

Answer 1

您正在尝试搜索 string在 filename , 使用 open(filename, 'r').read() :

import os

user_input = input('What is the name of your directory')
directory = os.listdir(user_input)

searchstring = input('What word are you trying to find?')

for fname in directory:
    if os.path.isfile(user_input + os.sep + fname):
        # Full path
        f = open(user_input + os.sep + fname, 'r')

        if searchstring in f.read():
            print('found string in file %s' % fname)
        else:
            print('string not found')
        f.close()

我们使用 user_input + os.sep + fname获取完整路径。 os.listdir给出文件和目录名称，所以我们使用 os.path.isfile检查文件。