Day 14 Homework2

创建登录-注册模块

import os
library = open('./library.txt', "a+")

reminder1 = """
-----------------------------------
1. 登录
2. 注册
-----------------------------------
"""

print(reminder1)

choice = ""

while choice == "":
    choice = input('请输入选择：')

choice = int(choice)

account_login = password_login = account_reg = password_reg = ""


if choice == 1:
    while account_login == "":
        account_login = input('请输入账号：')
    while password_login == "":
        password_login = input('请输入密码：')

    accnt_pwd_log = (account_login, password_login)

    library.seek(0)
    dict_lib = dict(list(eval(library.read())))

    if account_login in dict_lib:
        if password_login == dict_lib[account_login]:
            print("登录成功")
        else:
            print("登陆失败，密码不正确")
    else:
        print('登录失败，账号不存在')


if choice == 2:
    while account_reg == "":
        account_reg = input('请输入账号：')
    while password_reg == "":
        password_reg = input('请输入密码：')

    accnt_pwd_reg = (account_reg, password_reg)

    library.write(f'{accnt_pwd_reg}, ')

    print('请登录')