# 登錄注冊退出選擇界面


def main_page():
    print('=' * 40)
    print("**歡迎來到學生管理系統**".center(30, " "))
    print("★1.登     錄".center(36, " "))
    print("★2.注     冊".center(36, " "))
    print("★3.退     出".center(36, " "))
    print('=' * 40)
    num_1 = int(input("請輸入(1 - 3):"))
    return num_1


# 注冊界面


def register():
    print("歡迎來到注冊頁面")
    with open(r'student_information', 'r', encoding='utf - 8')as f:
        users = eval(f.read())
        user_name = input("用戶名:")
        pw = input("密碼:")
        list_2 = list(users)
        if user_name in list_2:
            print("此賬號已經注冊過了，請重新輸入賬號")
            register()
        else:
            users[user_name] = pw
    with open(r'student_information', 'w', encoding='utf-8')as f:
        f.write(str(users))
        value = input('是否繼續(xù)(y/n):')
        if value == 'y':
            register()
        if value == 'n':
            main_page()


# 登錄界面
def sign_in():
    print("歡迎來到登錄界面")
    user_name = input("請輸入學生姓名:")
    user_pw = input("請輸入密碼:")
    with open(r'student_information', encoding='utf - 8') as f:
        line = eval(f.read())
    list_1 = list(line)
    if len(list_1) == 0:
        print("請先注冊")
        main_page()
    else:
        if user_name in list_1:
            if line[user_name] == user_pw:
                print("登錄成功")

                main_page_1()
            else:
                print("密碼錯誤")
                sign_in()

        else:
            print("賬號錯誤")
            sign_in()
    return user_name


def main_page_1():

    print("==" * 30)
    print("               歡迎來到學生管理系統1.0")
    print("                 1.添加學生的信息")
    print("                 2.刪除學生的信息")
    print("                 3.修改學生的信息")
    print("                 4.查詢學生的信息")
    print("                 5.遍歷所有學生的信息")
    print("                 6.退出系統")
    print("==" * 30)


# 執(zhí)行函數


num_2 = main_page()

# 登錄管理
if num_2 == 1:
    sign_in()

# 注冊
if num_2 == 2:
    register()
# 退出
else:
    exit()