编写移除文本文件许可证的Python函数时遇变量未定义错误求助

错误根源 #

报错"local variable 'fout' referenced before assignment"的核心原因：

• 当打开新文件newfile_name失败时，try块里的fout = open(...)没执行，fout变量根本没被定义，但后续代码依然调用fout.write()和fout.close()，直接触发未定义错误。
• 加global fout完全没用，因为就算把fout声明为全局变量，异常分支里也没给它赋值，后续使用依然会报错。

另外代码还有几个潜在问题：

• 如果原文件file_name打开失败，后续循环for line in fin会直接报错；
• 若文件中找不到"*** START OF"或"*** END OF"，text_start和text_end会未定义，触发另一个错误；
• 用r+模式打开新文件，若文件不存在会直接失败，应该用w模式（创建新文件或覆盖已有文件）；
• 手动调用close()不够安全，一旦中间代码报错，文件可能无法正常关闭。

修复后的代码 #

def antipreamble(file_name):
    """Removes the preamble from a text file"""
    # 初始化标记变量，避免未定义
    text_start = -1
    text_end = -1

    # 打开原文件，用with自动管理关闭
    try:
        with open(file_name, "r") as fin:
            print(f"Opened {file_name} successfully")
            # 遍历文件找起止标记
            for i, line in enumerate(fin):
                if "*** START OF" in line:
                    text_start = i
                if "*** END OF" in line:
                    text_end = i
    except Exception as e:
        print(f"Sorry - could not open {file_name}: {str(e)}")
        return  # 打开失败直接返回，不执行后续逻辑

    # 检查是否找到起止标记
    if text_start == -1 or text_end == -1:
        print("Error: Could not find '*** START OF' or '*** END OF' in the file")
        return

    newfile_name = file_name[:-4] + "_new.txt"
    # 打开新文件写入，用w模式自动创建文件
    try:
        with open(newfile_name, "w") as fout:
            print(f"Opened {newfile_name} successfully")
            # 重新读取原文件，写入中间内容
            with open(file_name, "r") as fin:
                for i, line in enumerate(fin):
                    if text_start < i < text_end:
                        fout.write(line)
    except Exception as e:
        print(f"Sorry - could not write to {newfile_name}: {str(e)}")
        return

    print(f"Preamble removed successfully, saved to {newfile_name}")

关键修复点 #

• 用with语句管理文件，无需手动调用close()，自动处理文件关闭；
• 打开文件失败后直接return，避免后续代码使用未定义的文件对象；
• 初始化text_start和text_end为-1，后续检查是否找到标记；
• 写入新文件用w模式，确保文件不存在时自动创建；
• 用enumerate()替代手动计数i，代码更简洁；
• 捕获异常时打印具体错误信息，便于排查问题。

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