Day12_作业_学生管理系统登录注册界面

def register_stu():
    f = open("users.txt", 'r+')
    users = eval(f.read()) #f.read()读取的是字符串，用eval()将字符串转化为字典
    while True:
        account = input('请输入账号： ')
        passwd = input('请输入密码： ')
        if len(account) != 0 and account not in users and len(passwd) != 0:
            users[account]= {'passwd':passwd} #往字典中插入新数据
            f.seek(0)
            f.truncate()  #清空文件内容
            f.write(str(users)) #将字典写入文件
            print('恭喜，注册成功')
            f.close()
            break
        elif len(account) == 0:
            print('用户名不能为空，请重新输入。')
        elif account in users:
            print('用户名重复，请重新输入。' )
        elif len(passwd) == 0:
            print('密码不能为空，请重新输入。' )


def sign_stu():
    f = open("users.txt", 'r+')
    users = eval(f.read()) #f.read()读取的是字符串，用eval()将字符串转化为字典
    while True:
        account = input('请输入账号，若无账号，请输入q回到注册界面： ')
        if account == 'q':
            break
        passwd = input('请输入密码： ')
        if len(account) == 0:
            print('用户名不能为空，请重新输入。')
        elif account not in users:
            print('用户名不存在，请重新输入。' )
        elif len(passwd) == 0:
            print('密码不能为空，请重新输入。' )
        elif account in users and passwd == users[account]['passwd']:
            print('登录成功')
            f.close()
            return 1
        else:
            print('密码输入错误，请重新输入')



while True:

    print('========================================================')
    print('❀❀❀欢迎来到学生管理系统：')
    print('    ')
    print('          ❤ 1.  登录')
    print('          ❤ 2.  注册')
    print('          ❤ 3.  退出')
    print('    ')
    print('========================================================')
    number = int(input('请选择（1-3）：'))

    if number == 2:
        register_stu()
    elif number == 3:
        break
    elif number == 1:
        if sign_stu() == 1:
            information_stu()