'''
==================
1.      注冊
2.      登錄
==================
'''
'''
文件中內(nèi)容匪凉，文件名 register
[{'user_name': 'lihaowen', 'passwd': '123456', 'id': 'python1902001'}, {'user_name': 'xiaogou', 'passwd': '234567', 'id': 'python1902002'}, 
{'user_name': 'wang', 'passwd': '123456', 'id': 'python1902003'}, {'user_name': 'lihuahua', 'passwd': '123456', 'id': 'python1902004'}]
'''
def page():
    print("========================================")
    print('歡迎：')
    print('\n')
    print('                    ','*1','注冊')
    print('                    ','*2','登錄')
    print('                    ','*5','返回')
    print("========================================")


def register():
    user_name = input("請輸入用戶名:")
    print(user_name)
    with open('register','r+') as f:
        data = eval(f.read())
        # print(data,type(data))
        for i in data:
            print(i['user_name'])
            if user_name in i['user_name']:
                print("用戶名已存在")
                register()
        else:
            passwd = input("請輸入密碼:")
            id = 'python19020' + '0' * (2 - len(str(int(data[len(data) - 1]['id'][-2:]) + 1))) + str(
                int(data[len(data) - 1]['id'][-2:]) + 1)
            data_new = {'user_name':user_name,'passwd':passwd,'id':id}
            # print(data_new)
            data.append(data_new)
            print(data)
            f.seek(0)
            f.write(str(data))
            print("注冊成功")
# register()
def enter():
    user_name = input("請輸入用戶名:")
    # print(user_name)
    with open('register', 'r+') as f:
        data = eval(f.read())
        # print(data,type(data))
        for i in data:
            print(i['user_name'])
            if user_name in i['user_name']:
                print(i)
                print(i['passwd'])
                passwd = input("請輸入密碼:")
                if i['passwd'] == passwd:
                    print("登陸成功")
                    main_page()
                else:
                    print("密碼錯誤")
                    enter()
            else:
                print("沒有該用戶")
                main_page()

def main_page():
    page()
    choice = int(input("輸入對應數(shù)字選擇功能："))
    if choice == 1:
        register()
        main_page()
    elif choice == 2:
        enter()
        main_page()
    elif choice == 5:
        page()
        main_page()
main_page()