Python第五次作业

#制作一个注册登录模块
#注册要将用户填入的账户和密码保存到一个文件中（user.bin）
#登录：将用户填入账户密码和user.bin中保存的账户密码进行比对，如果账户和密码完全相同，那么登录成功，否则登录失败

UI = """
用户登录系统
1.注册
2.登录
3.退出
"""
#读取users.bin
def  load(path):
    return json.load(open(path, "rt"))


# 保存users.bin
def save(dic, path: str):
    json.dump(dic,open(path, "wt"))

while True:
    input("按回车继续...")
    print(UI)
    num = input("请输入你的选择")
    if num == "1":
        #从users.bin读取已有的账户数据，用户注册完以后，将新账户注册到已有数据中
        if os.path.exists("users.bin"):
            dc = load("users.bin")
        else:
            dc = {}
        # print("进入到注册环节")
        account = input("请输入账户")
        passwd = input("请输入密码")
        confirm = input("请确认密码")
        if dc.get(account):
            print("账户已存在请重新输入")
        elif passwd == confirm:
            md5 = hashlib.md5(passwd.encode())
            encrypt = md5.hexdigest()
            dc[account] = encrypt
            save(dc, path="users.bin")
            print("注册成功")
    elif num == "2":
        account = input("请输入账户")
        pwd = input("请输入密码")
        dc = load("users.bin")
        if dc.get(account):
            md5 = hashlib.md5(pwd.encode())
            encrypt_pwd = md5.hexdigest()
            if dc[account] == encrypt_pwd:
                print("登录成功")
            else:
                print("登录失败")
        else:
            print("账户或者密码错误")
        print("进入到登陆环节")
    else:
        sys.exit()