import json


student_data = []
def add_student():
    id = input("请输入学员学号: ")
    student_id = input("请输入学员姓名: ")
    gender = input("请输入性别: ")
    tel = input("请输入电话号码：")

    student_info = {
        "学号": id,
        "姓名": student_id,
        "性别": gender,
        "电话号码":tel
    }

    student_data.append(student_info)
    print("学员信息添加成功！")



def delete_student():
    student_id = input("请输入要删除的学号: ")

    for student in student_data:
        if student["学号"] == student_id:
            student_data.remove(student)
            print("学员信息删除成功！")
            break
    else:
        print("未找到该学号的学员信息。")



def modify_student():
    student_id = input("请输入要修改的学号: ")

    for student in student_data:
        if student["学号"] == student_id:
            name = input("请输入新的姓名: ")
            age = input("请输入新的性别: ")
            tel = input("电话号码: ")
            student["姓名"] = name
            student["性别"] = age
            student["电话号码"] = tel
            print("学员信息修改成功！")
            break
    else:
        print("未找到该学号的学员信息。")


def select():
    student_id = input("请输入要修改的学号: ")

    for student in student_data:
        if student["学号"] == student_id:
            print(student["学号"],student["姓名"],student["性别"],student["电话号码"])
            break
    else:
        print("未找到该学号的学员信息。")

# 定义函数：保存学员信息到文件
def save_student_data():
    with open("student_data.json", "w") as file:
        json.dump(student_data, file)
    print("学员信息已保存！")



def load_student_data():
    try:
        with open("student_data.json", "r") as file:
            data = json.load(file)
            student_data.extend(data)
            print("学员信息已加载！")
    except FileNotFoundError:
        print("未找到学员信息文件。")


# 主程序
load_student_data()

while True:
    print("\n学生信息管理系统")
    print("1. 添加学员信息")
    print("2. 删除学员信息")
    print("3. 修改学员信息")
    print("4. 保存学员信息")
    print("5. 查询学员信息")
    print("6. 显示学员信息")
    print("7. 退出系统")
    choice = input("请输入操作序号: ")

    if choice == "1":
        add_student()
    elif choice == "2":
        delete_student()
    elif choice == "3":
        modify_student()
    elif choice == "4":
        save_student_data()
    elif choice == "5":
        select()
    elif choice == "6":
        print("学员信息如下:")
        for student in student_data:
            print("学号:", student["学号"])
            print("姓名:", student["姓名"])
            print("性别:", student["性别"])
            print("电话号码:", student["电话号码"])
            print("--------------------------")
    elif choice == "7":
        save_student_data()
        print("感谢使用学生信息管理系统，再见！")
        break
    else:
        print("无效的操作序号，请重新输入。")