设计一个注册登录系统

1.注册功能

将用户注册的账户密码保存在account.txt文件中
账户:密码(经过hashlib加密)

2.登陆功能

当用户输入的账户和密码与account.txt中保存的账户密码一样则登陆成功
如果账户不同则打印账户错误
如果密码不同则打印密码错误

import hashlib


def register():
    account = input("请输入账户：")
    password = input("请输入密码：")

    # 使用 hashlib 对密码进行加密
    password_hash = hashlib.sha256(password.encode()).hexdigest()

    # 将账户和加密后的密码保存在 account.txt 文件中
    with open("account.txt", "a") as file:
        file.write(f"{account}:{password_hash}\n")
    print("注册成功！")


def login():
    account = input("请输入账户：")
    password = input("请输入密码：")

    # 使用 hashlib 对输入的密码进行加密
    password_hash = hashlib.sha256(password.encode()).hexdigest()

    # 读取 account.txt 文件中的账户密码信息
    with open("account.txt", "r") as file:
        accounts = file.readlines()

    for account_info in accounts:
        stored_account, stored_password_hash = account_info.strip().split(":")

        # 检查输入的账户是否匹配
        if account == stored_account:
            # 检查输入的密码是否匹配
            if password_hash == stored_password_hash:
                print("登录成功！")
                return

    print("账户或密码错误！")


# 主程序
menu = """
1.注册
2.登录

请选择功能："""

select = int(input(menu))

if select == 1:
    register()
elif select == 2:
    login()
else:
    print("无效的选择")