Python如何在所有文件中搜索字符串及代码总返回True问题排查

代码问题分析 #

• 逻辑重复冗余：外层循环已经打开了文件逐行遍历，却在每行遍历中额外调用check函数重新读取整个文件判断字符串是否存在，完全不需要双重读取文件。只要文件中存在一次目标字符串，所有行的check调用都会返回True，自然输出全为True。
• 未过滤目录：os.listdir()会返回当前目录下的所有条目，包括子文件夹，直接对文件夹执行打开操作会触发IO错误。
• 逻辑不符合需求：你的需求是全局判断字符串是否在目录下的文件中存在，但现有代码是对每个文件的每一行都输出一次判断结果，完全偏离需求。
• 没有支持用户输入：目前目标字符串是写死的text = 'T'，没有留用户输入的入口。

修复后代码 #

如果你的需求是判断目标字符串是否存在于当前目录任意一个普通文件中，存在则执行X，否则执行Y，参考代码如下：

import os

# 支持用户输入目标字符串
text = input("请输入要查找的字符串：")

def check_file_has_text(filename, target):
    # 兼容编码问题，避免特殊文件报错
    with open(filename, 'r', encoding='utf-8', errors='ignore') as f:
        return any(target in line for line in f)

has_text = False
# 遍历当前目录所有条目
for entry in os.listdir(os.getcwd()):
    entry_path = os.path.join(os.getcwd(), entry)
    # 只判断普通文件，跳过目录、软链接等非文件条目
    if os.path.isfile(entry_path):
        if check_file_has_text(entry_path, text):
            has_text = True
            # 找到后可以提前退出，不需要扫描剩余文件
            break

if has_text:
    # 存在则执行操作X
    print("找到目标字符串，执行操作X")
else:
    # 所有文件都不存在则执行操作Y
    print("未找到目标字符串，执行操作Y")

如果你的需求是判断字符串在当前目录所有普通文件中都存在，全部存在才执行X，否则执行Y，仅需替换循环判断部分的逻辑：

all_file_has_text = True
for entry in os.listdir(os.getcwd()):
    entry_path = os.path.join(os.getcwd(), entry)
    if os.path.isfile(entry_path):
        if not check_file_has_text(entry_path, text):
            all_file_has_text = False
            break

if all_file_has_text:
    print("所有文件都存在目标字符串，执行操作X")
else:
    print("存在文件不含目标字符串，执行操作Y")

内容的提问来源于stack exchange，提问作者TS Pham