登錄系統(tǒng) （函數(shù)封裝）

import File_operation


def register():
    while True:
        reg_id = input('請輸入注冊名字(3-6位)：')
        if 3 < len(reg_id) < 6:
            break
        else:
            print('注冊有誤，請重試')
    while True:
        reg_pa = input('請輸入密碼(6-12位)：')
        if 6 < len(reg_pa) < 12:
            break
        else:
            print('密碼輸入有誤，請重試')

    res = File_operation.file_json_r('file/stu_info.json')
    if not res:
        res = {}
    elif reg_id in res:
        print('這個ID 已經(jīng)被注冊')
        return
    else:
        res[reg_id] = reg_pa
        File_operation.file_json_w('file/stu_info.json', res)
        print('注冊成功')


def login():
    name = input('請輸入你的名字：')
    pass_word = input('請輸入你的密碼：')
    re = File_operation.file_json_r('stu_info.json')
    if name in re:
        if re[name] == pass_word:
            print('登錄成功')
        else:
            print('密碼錯誤')
    elif not re:
        print('登錄失敗，賬號沒有注冊')
    else:
        print('登錄失敗')


def show_main_page():
    while True:
        re = File_operation.file_open('file/login_interface')
        print(re)
        choose = input('Please select number:')
        if choose == '1':
            login()
        elif choose == '2':
            register()
        else:
            break

if __name__ == '__main__':
    show_main_page()