Python遍历文件行仅替换首个匹配的用户输入如何实现？

你现有代码存在两个核心问题：

• 调用str.replace()时没有指定替换次数，默认会替换一行内所有匹配内容，且遍历所有行时都执行替换操作，导致全局匹配全部被替换
• 第二个for lines in old_file循环完全不会执行：old_file.readlines()已经读完了整个文件的内容，文件指针已经到末尾，后续遍历拿不到任何内容，这部分逻辑是无效的

from tempfile import mkstemp
from shutil import move, copymode
from os import fdopen, remove

def replace(file_path, pattern, subst):
    fh, abs_path = mkstemp()
    # 增加标记位，记录是否已经完成首次替换
    replaced = False
    with fdopen(fh,'w') as new_file:
        with open(file_path) as old_file:
            for line in old_file:
                if not replaced:
                    # 替换当前行最多1次匹配内容
                    new_line = line.replace(pattern, subst, 1)
                    new_file.write(new_line)
                    # 检查当前行是否发生了替换，如果是则更新标记
                    if new_line != line:
                        replaced = True
                else:
                    # 已经完成首次替换，后续行直接写入原内容
                    new_file.write(line)
    copymode(file_path, abs_path)
    remove(file_path)
    move(abs_path, file_path)

test_input = 'fruits'
replace_with = 'bread'
# 此处file_path仅作示例，请替换为实际文件路径
replace(file_path, test_input, replace_with)

• 新增replaced布尔标记，初始值为False，用于记录是否已经完成首次匹配替换
• 不需要提前读取所有行到内存，直接逐行遍历原文件即可，大文件场景下内存占用更低
• 未完成首次替换时，调用line.replace(pattern, subst, 1)，第三个参数1表示仅替换当前行内第一个匹配项
• 检测到当前行发生替换后，将replaced设为True，后续所有行直接写入原内容，不再做替换处理

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