Python:创建/写入文件直到循环结束

Question

我正在尝试使用 Python 将用户输入的所有内容保存在一个文本文件中。我想确保所有输入的输入都存储在文件中，直到我完全退出程序，在这种情况下，直到我按“enter”停止列表。我还需要检查输入的名称，看看它是否与之前的任何条目匹配。

我的程序现在的问题是，当我退出代码时，文本文件会更新输入的最新名称。我需要我的程序将所有这些名称保存在列表中，直到程序结束，因为我必须确保没有重复项。我将不得不警告用户该名称已经存在，我也需要帮助。我在下面的代码中创建了一个单独的函数来根据我的输入创建和写入文本文件，但我也注意到我可以在 get_people() 函数中实现它。我不确定是否为它创建新功能的最佳策略是什么。写文件肯定有问题。

文本文件应具有以下格式:

Taylor

Selena

Martha

Chris

下面是我的代码:

def get_people():
    print("List names or <enter> to exit")
    while True:
        try:
            user_input = input("Name: ")
            if len(user_input) > 25:
                raise ValueError
            elif user_input == '':
                return None
            else:
                input_file = 'listofnames.txt'
                with open(input_file, 'a') as file:
                    file.write(user_input + '\n')
                return user_input

        except ValueError:
            print("ValueError! ")


# def name_in_file(user_input):
#     input_file = 'listofnames.txt'
#     with open(input_file, 'w') as file:
#             file.write(user_input + '\n')
#     return user_input


def main():
    while True:
    try:
        user_input = get_people()
        # name_in_file(user_input)
        if user_input == None:
            break

    except ValueError:
        print("ValueError! ")

main()

Answer 1

问题在于代码打开文件的方式:

with open(input_file, 'w') as file:

检查手册 - https://docs.python.org/3.7/library/functions.html?highlight=open#open由于 "w"，代码每次 open() 都会覆盖文件。它需要打开它以附加 "a":

with open(input_file, 'a') as file:

如果文件不存在，追加将创建该文件，或者追加到任何现有的同名文件的末尾。

编辑:要检查您是否已经看到这个名字，将“已经看到”的名字列表传递给 get_people() 函数，并将任何新名字也附加到该列表。

def get_people( already_used ):
    print("List names or <enter> to exit")
    while True:
        try:
            user_input = input("Name: ")
            lower_name = user_input.strip().lower()
            if len(user_input) > 25:
                raise ValueError
            elif lower_name in already_used:
                print("That Name has been used already")
            elif user_input == '':
                return None
            else:
                already_used.append( lower_name )
                input_file = 'listofnames.txt'
                with open(input_file, 'a') as file:
                    file.write(user_input + '\n')
                return user_input

        except ValueError:
            print("ValueError! ")

def main():
    already_used = []
    while True:
        try:
            user_input = get_people( already_used )
            # name_in_file(user_input)
            if user_input == None:
                break

        except ValueError:
            print("ValueError! ")

main()