使用文本文件的 Python 登录和注册系统

Question

嘿，我正在尝试使用文本文件创建一个系统，用户可以在其中注册和登录。所有数据都将以纯文本形式存储在名为 User_Data.txt 的文本文件中。 我的代码有效，但我想知道我是否遗漏了什么或者我是否可以以任何方式改进它。 抱歉，提前格式化错误代码。

def choices():
    print("Please choose what you would like to do.")
    choice = int(input("For Sigining Up Type 1 and For Signing in Type 2: "))
    if choice == 1:
       return getdetails()
    elif choice == 2:
       return checkdetails()
    else:
       raise TypeError

def getdetails():
    print("Please Provide")
    name = str(input("Name: "))
    password = str(input("Password: "))
    f = open("User_Data.txt",'r')
    info = f.read()
    if name in info:
        return "Name Unavailable. Please Try Again"
    f.close()
    f = open("User_Data.txt",'w')
    info = info + " " +name + " " + password
    f.write(info)

def checkdetails():
    print("Please Provide")
    name = str(input("Name: "))
    password = str(input("Password: "))
    f = open("User_Data.txt",'r')
    info = f.read()
    info = info.split()
    if name in info:
        index = info.index(name) + 1
        usr_password = info[index]
        if usr_password == password:
            return "Welcome Back, " + name
        else:
            return "Password entered is wrong"
    else:
        return "Name not found. Please Sign Up."

print(choices())

Answer 1

您可以做很多改进。 首先，将功能拆分为更小的功能。

PASSWORD_FNAME = "User_Data.txt"

def get_existing_users():
    with open("r", PASSWORD_FNAME ) as fp:
         for line in fp.readlines():
             # This expects each line of a file to be (name, pass) seperated by whitespace
             username, password = line.split()
             yield username, password

def is_authorized(username, password):
    return any((user == (username, password) for user in get_existing_users()) 

def user_exists(username):
    return any((usr_name == username) for usr_name, _ in get_existing_users())
    # above is equivalent of:
    #
    # for usr_name, _ in get_existing_users():
    #     if usr_name == username:
    #        return True
    # return False

def ask_user_credentials():
    print("Please Provide")
    name = str(input("Name: "))
    password = str(input("Password: "))
    return name, password

def checkdetails():
    name, password = ask_user_credentials()
    if is_authorized(name, password):
       return "Welcome Back, " + name
    if user_exists(name):
       return "Password entered is wrong"
    return "Name not found. Please Sign Up."

def getdetails():
    name, password = ask_user_credentials()
    if not user_exists(name):
       return "Name Unavailable. Please Try Again"
    # Not sure tho what would You like to do here

最好记住在阅读文件时始终关闭文件。 因此，如果您执行类似以下操作： f = open("r", "file.txt")请记住稍后始终调用f.close() 。 如果您使用上下文管理器并这样做：

with open("r", "file.txt") as fp:
     print(fp.read())

它会在最后自动为您关闭文件。

Answer 2

首先，修复int(input("For Sigining Up Type 1")处的拼写错误。除此之外，我会添加一些目的，例如存储密码或其他内容。

Answer 3

例如，您可以使用简单的密码恢复系统扩展您的脚本。 我觉得学习一下还是有用的。。。

您可以实施一种简单的哈希系统，以避免将密码保存为纯文本。

如果要添加 GUI，请考虑使用 Tkinter。 https://docs.python.org/3/library/tkinter.html

让我们知道。

祝你好运，继续编码 <3