import json


def login_interface():
    """
    登錄界面
    :return: 無
    """
    while True:
        print("登錄注冊：")
        print('='.center(40, '='))
        print("1.登錄\n2.注冊\n3.任意鍵退出")
        select1 = input("請輸入：")
        print('='.center(40, '='))
        if select1 == '1':
            login()
        if select1 == '2':
            register()
        if not (select1 == '1' or select1 == '2'):
            break


def login():
    """
    登錄入口
    :return: 無
    """
    while True:
        print('='.center(40, '='))
        name_str = input("輸入賬號：")
        password_str = input("輸入密碼：")
        try:
            with open("users.json", 'r', encoding='utf-8') as f:
                content = f.read()
                try:
                    users_list = json.loads(content)
                except json.decoder.JSONDecodeError:
                    users_list = []
                users_count = len(users_list)
                count = 0
                for user_dict1 in users_list[:]:
                    if name_str == user_dict1['賬號'] and password_str == user_dict1['密碼']:
                        print("登錄成功！")
                        stu_sys()
                        return
                    if count == users_count - 1:
                        print("登錄失斶制摺恰起！")
                        break
                    count += 1
                if len(users_list) == 0:
                    print("登錄失敗!")
        except FileNotFoundError:
            print("登錄失敗!")
        print('='.center(40, '='))
        print("1.繼續(xù)\n2.返回")
        select1 = input("請選擇：")
        while True:
            if select1 == '1':
                break
            if select1 == '2':
                return
            print("輸入錯誤！")
            print("1.繼續(xù)\n2.返回")
            select1 = input("請選擇：")


def register():
    """
    注冊入口
    :return: 無
    """
    while True:
        print('='.center(40, '='))
        user_dict = {}
        name_str = input("輸入賬號：")
        password_str = input("輸入密碼：")
        try:
            with open("users.json", 'r', encoding='utf-8') as f:
                content = f.read()
                try:
                    users_list = json.loads(content)
                except json.decoder.JSONDecodeError:
                    users_list = []
                users_count = len(users_list)
                count = 0
                for user_dict1 in users_list[:]:
                    if name_str == user_dict1['賬號']:
                        print("注冊失斚ㄊ亍！")
                        break
                    if count == users_count - 1:
                        user_dict['賬號'] = name_str
                        user_dict['密碼'] = password_str
                        users_list.append(user_dict)
                        print("注冊成功！")
                        break
                    count += 1
                if len(users_list) == 0:
                    user_dict['賬號'] = name_str
                    user_dict['密碼'] = password_str
                    users_list.append(user_dict)
                    print("注冊成功而柑！")
        except FileNotFoundError:
            users_list = []
            user_dict['賬號'] = name_str
            user_dict['密碼'] = password_str
            users_list.append(user_dict)
            print("注冊成功移国！")
        with open("users.json", 'w', encoding='utf-8') as f:
            f.write(json.dumps(users_list))
        print('='.center(40, '='))
        print("1.繼續(xù)\n2.返回")
        select1 = input("請選擇：")
        while True:
            if select1 == '1':
                break
            if select1 == '2':
                return
            print("輸入錯誤吱瘩！")
            print("1.繼續(xù)\n2.返回")
            select1 = input("請選擇：")


def interface():
    """
    歡迎界面
    :return: 無
    """
    print('='.center(40, '='))
    print("歡迎ZH:")
    print(' '*15 + "1. 添加學(xué)生")
    print(' '*15 + "2. 查看學(xué)生")
    print(' '*15 + "3. 修改學(xué)生信息")
    print(' '*15 + "4. 刪除學(xué)生")
    print(' '*15 + "5. 返回")
    print('='.center(40, '='))


def add_stu():
    """
    添加學(xué)生信息
    :return: 無
    """
    stu_file = open("D:\\學(xué)生.txt", 'a+')
    while True:
        print('='.center(40, '='))
        stu_name = input("請輸入學(xué)生姓名：")
        stu_age = int(input("請輸入學(xué)生的年齡："))
        stu_tel = input("請輸入學(xué)生的電話：")
        stu_file.write('姓名:' + stu_name + ' 年齡:%d' % stu_age + ' 電話:' + stu_tel + '\n')
        print("添加成功!")
        print('='.center(40, '='))
        while True:
            print("1.繼續(xù)\n2.返回")
            n = input("請輸入選擇(1-2):")
            if n == '1':
                break
            if n == '2':
                stu_file.close()
                return
            print("輸入錯誤！")


def view_stu():
    """
    查看學(xué)生信息
    :return: 無
    """
    stu_file = open("D:\\學(xué)生.txt")
    while True:
        print('='.center(40, '='))
        stu_name = input("請輸入查看學(xué)生的姓名：")
        flag = 0
        for stu_str in stu_file.readlines():
            name_str = ''
            for ch in stu_str[3:]:
                if ch == ' ':
                    break
                name_str += ch
            if name_str == stu_name:
                print(stu_str[:-2])
                flag = 1
                break
        if flag == 0:
            print("很遺憾迹缀，未找到該學(xué)生信息使碾！")
        print('='.center(40, '='))
        while True:
            print("1.繼續(xù)\n2.返回")
            n = input("請輸入選擇(1-2):")
            if n == '1':
                stu_file.seek(0)
                break
            if n == '2':
                stu_file.close()
                return
            print("輸入錯誤！")


def modify_stu():
    """
    修改學(xué)生信息
    :return: 無
    """
    while True:
        print('='.center(40, '='))
        stu_name = input("請輸入修改學(xué)生的姓名：")
        flag = 0
        file_data = ""
        with open("D:\\學(xué)生.txt", 'r') as stu_file:
            for stu_str in stu_file.readlines():
                name_str = ''
                for ch in stu_str[3:]:
                    if ch == ' ':
                        break
                    name_str += ch
                if name_str == stu_name:
                    stu_age = int(input("請輸入修改學(xué)生的年齡："))
                    stu_tel = input("請輸入修改學(xué)生的電話：")
                    stu_str = '姓名:' + stu_name + ' 年齡:%d' % stu_age + ' 電話:' + stu_tel + '\n'
                    flag = 1
                file_data += stu_str
        with open("D:\\學(xué)生.txt", 'w') as stu_file:
            stu_file.write(file_data)
        if flag == 0:
            print("很遺憾祝懂，未找到該學(xué)生信息票摇！")
        if flag == 1:
            print("修改成功！")
        print('='.center(40, '='))
        while True:
            print("1.繼續(xù)\n2.返回")
            n = input("請輸入選擇(1-2):")
            if n == '1':
                break
            if n == '2':
                return
            print("輸入錯誤砚蓬！")


def remove_stu():
    """
    刪除學(xué)生信息
    :return: 無
    """
    while True:
        print('='.center(40, '='))
        stu_name = input("請輸入刪除學(xué)生的姓名：")
        flag = 0
        file_data = ""
        with open("D:\\學(xué)生.txt", 'r') as stu_file:
            for stu_str in stu_file.readlines():
                name_str = ''
                for ch in stu_str[3:]:
                    if ch == ' ':
                        break
                    name_str += ch
                if name_str == stu_name:
                    flag = 1
                    continue
                file_data += stu_str
        with open("D:\\學(xué)生.txt", 'w') as stu_file:
            stu_file.write(file_data)
        if flag == 0:
            print("很遺憾矢门，未找到該學(xué)生信息！")
        if flag == 1:
            print("刪除成功灰蛙！")
        print('='.center(40, '='))
        while True:
            print("1.繼續(xù)\n2.返回")
            n = input("請輸入選擇(1-2):")
            if n == '1':
                break
            if n == '2':
                return
            print("輸入錯誤祟剔！")


def stu_sys():
    """
    學(xué)生系統(tǒng)
    :return: 無
    """
    while True:
        interface()
        n = input("請輸入選擇(1-5):")
        while True:
            if n == '1':
                add_stu()
            if n == '2':
                view_stu()
            if n == '3':
                modify_stu()
            if n == '4':
                remove_stu()
            if n == '5':
                return
            if not (1 <= int(n) <= 5):
                print("輸入錯誤！")
                n = input("請輸入選擇(1-5):")
                continue
            interface()
            n = input("請輸入選擇(1-5):")


login_interface()