Replit平台Python函数多次调用仅首次生效问题排查求助

问题根源 #

• 全局声明的allNames是文件迭代器，第一次调用findName时已遍历完所有内容，文件指针停在末尾，后续再遍历该对象不会返回任何数据，导致第二次及之后的函数调用完全失效。
• 全局打开的文件未关闭，会占用系统资源，可能干扰后续代码对文件或相关资源的操作。

修复代码 #

将文件打开逻辑移入函数内部，用with语句自动管理文件生命周期，同时优化去重逻辑：

import string

def findName(name, outputFile):
    # 每次调用重新打开源文件，确保从开头读取
    with open("allNames.csv", "r") as allNames, open(f"{outputFile}.csv", "w") as outfile:
        outfile.write("Artist \tSong \tYear\n")
        alreadyAdded = set()  # 集合比列表的成员检查效率更高
        for aline in allNames:
            fields = aline.split("\t")
            # 用strip()处理换行符和空白，避免匹配失败
            if fields[-1].strip() == name.strip():
                dataline = f"{fields[0]}\t{fields[1]}\t{fields[3]}"
                if dataline not in alreadyAdded:
                    outfile.write(f"{dataline}\n")
                    alreadyAdded.add(dataline)

# 现在可正常多次调用
findName("Mary Anne", "mary anne")
findName("Jack", "jack")
findName("Mary", "mary")
findName("Peter", "peter")

优化说明 #

• with语句会自动关闭文件，无需手动调用close()，避免资源泄漏。
• strip()处理字符串两端空白（含换行符），解决因文件行尾换行符不一致导致的匹配问题。
• 用集合存储已添加内容，将成员检查的时间复杂度从O(n)降至O(1)，大数据量下性能更优。

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