Day13-作业

Day13-作业

print('===============================')
print('    **欢迎来到时间管理系统**     ')
print()
print('       1. 登   录              ')
print('       2. 注   册              ')
print('       3. 退   出              ')
print()
print('===============================')


def detection():
    f = open(r'files\user_date.txt', 'r', encoding='utf-8')
    result = f.read()
    if not result:
        f = open(r'files\user_date.txt', 'w', encoding='utf-8')
        f.write('{}')
        f.close()
    f = open(r'files\user_date.txt', 'r', encoding='utf-8')
    # f.seek(0)
    result = eval(f.read())
    return result


def select(num):
    if num == 1:
        return n1()
    elif num == 2:
        return n2()
    else:
        print('无效输入')
        num = int(input('请输入你的选择：'))
        return select(num)


def n1():
    num = input('请输入账号：')
    password = input('请输入密码：')
    result = detection()
    while f'{num}' in result:
        if result[f'{num}'] == password:
            print('登录成功')
            break
        else:
            print('密码错误')
            password = input('请输入密码：')


def n2():
    num = input('请输入账号（3-6位）：')
    if 3 <= len(num) <= 6:
        password = input('请输入密码（6-12位）：')
        if 6 <= len(password) <= 12:
            result = detection()
            if f'{num}' not in result:
                result[f'{num}'] = password
                print('恭喜你注册成功')
            else:
                print('该账号已经注册过！')
                print('请选择后续操作:输入1直接登录 输入2重新注册')
                num1 = int(input('请输入你的选择：'))
                select(num1)
            f = open(r'files\user_date.txt', 'w', encoding='utf-8')
            f.write(f'{result}')
            f.close()
        else:
            print('密码格式错误！')
            # password = input('请重新输入密码（6-12位）：')
            return n2()
    else:
        print('账号格式错误！')
        num = input('请重新输入账号（3-6位）：')
        return n2()


n = int(input('请选择（1-3）：'))
if n == 1:
    n1()
elif n == 2:
    n2()
elif n == 3:
    print('退出')
else:
    print('输入错误')