注册管理系统

注册管理系统

def register():
    print('开始注册...')
    f1 = open('files/user.txt', encoding='utf-8')
    user_list = eval(f1.read())
    f1.close()
    username = input('请输入账号: ')
    password = input('请输入密码: ')
    add_dict = {'name': username, 'word': password}
    user_list.append(add_dict)
    f1 = open('files/user.txt', 'w+', encoding='utf-8')
    f1.write(str(user_list))
    for x in range(len(user_list)):
        if user_list[x]['name'] and user_list[x]['word']:
            if user_list[x]['name'] == username and user_list[x]['word'] == password:
                print('注册成功!')
                f1.close()
            else:
                print('注册失败!该账号已经注册过!')
                f1.close()
        else:
            print('注册失败!')
            f1.close()


def login():
    print('开始登陆...')
    username = input('请输入账号: ')
    password = input('请输入密码: ')
    user = {'name': username, 'word': password}
    f = open('files/user.txt', encoding='utf-8')
    result = f.read()
    if not result:
        print('请先注册!')
        register()
    all_user = eval(result)
    # all_user = []
    for x in range(len(all_user)):
        if all_user[x]['name'] == user['name'] and all_user[x]['word'] == user['word']:
            print('登录成功!')
            menu()
        elif all_user[x]['name'] == user['name'] and all_user[x]['word'] != user['word']:
            print('登录失败!')
            print('是否继续登录? 是/否(1/0) ')
            enter = int(input())
            if enter == 1:
                login()
            elif enter == 0:
                menu()
        elif all_user[x]['name'] == user['name'] and all_user[x]['word'] != user['word']:
            print('登录失败!')
            print('是否继续登录? 是/否(1/0) ')
            enter = int(input())
            if enter == 1:
                login()
            elif enter == 0:
                menu()
                login()
    else:
        print('登录失败!该账号不存在!')
        print('是否继续登录? 是/否(1/0) ')
        enter = int(input())
        if enter == 1:
            login()
        elif enter == 0:
            menu()
            login()
    f.close()


def menu():
    print('================================')
    print('      ** 欢迎到xxx管理系统 **      ')
    print('      ♥ 1、登          录')
    print('      ♥ 2、注          册')
    print('      ♥ 3、退          出')
    print('================================')
    num = int(input('请选择(1~3): '))
    if num == 1:
        login()
        menu()
    elif num == 2:
        register()
        menu()
    elif num == 3:
        print('退出!')
        return 0
        # exit()


menu()