def main():
    while True:
        welcome_login()
        num1 = int(input('請選擇1-3：'))
        if num1 == 1:
            register_login()
        elif num1 == 2:
            register()
        else:
            return


def welcome_login():
    print('***  歡迎來到XX管理系統(tǒng)   ***')

    print('??? 1. 登          錄')
    print('??? 2. 注          冊')
    print('??? 3. 退          出')

    print('================================================')


def register_login():
    with open('./stuinfo.txt', 'r', encoding='utf-8') as f:
        users = eval(f.read())
    user_name = (input('請輸入3-6位數(shù)賬號：'))
    pw = input('輸入密碼：')
    if user_name not in users:
        print('登錄失??！請先注冊賬號')
    if pw not in users[user_name]:
        print('登錄失敗,密碼錯誤！')

    return student_info



def register():
    with open('./stuinfo.txt', 'r', encoding='utf-8') as f:
        xxx = eval(f.read())
    users = {}
    user1 = input('輸入3-6位注冊賬號：')
    if user1 in xxx:
        print('此賬號已被注冊！')
        return
    pw1 = input('輸入賬號密碼：')
    users[user1] = pw1
    with open('./stuinfo.txt', 'w', encoding='utf-8') as f:
        f.write(str(users))


def student_info():
    with open('./stuinfo.txt', 'r', encoding='utf-8') as f:
        xx2 = eval(f.read())
    print('歡迎' + str(xx2))