用python写学生管理系统

class Student:
    def __init__(self, id, name, age, gender):
        self.id = id
        self.name = name
        self.age = age
        self.gender = gender

class StudentManagementSystem:
    def __init__(self):
        self.students = []

    def add_student(self):
        id = input("请输入学生ID: ")
        name = input("请输入学生姓名: ")
        age = int(input("请输入学生年龄: "))
        gender = input("请输入学生性别: ")
        student = Student(id, name, age, gender)
        self.students.append(student)
        print(f"添加学生 {name} 成功！")

    def delete_student(self):
        id = input("请输入要删除的学生ID: ")
        for student in self.students:
            if student.id == id:
                self.students.remove(student)
                print(f"删除学生 {student.name} 成功！")
                break
        else:
            print(f"未找到ID为 {id} 的学生。")

    def update_student(self):
        id = input("请输入要更新的学生ID: ")
        for student in self.students:
            if student.id == id:
                name = input("请输入新的学生姓名: ")
                age = int(input("请输入新的学生年龄: "))
                gender = input("请输入新的学生性别: ")
                student.name = name
                student.age = age
                student.gender = gender
                print(f"更新学生 {name} 成功！")
                break
        else:
            print(f"未找到ID为 {id} 的学生。")

    def search_student(self):
        id = input("请输入要查找的学生ID: ")
        for student in self.students:
            if student.id == id:
                print(f"ID: {student.id}, 姓名: {student.name}, 年龄: {student.age}, 性别: {student.gender}")
                break
        else:
            print(f"未找到ID为 {id} 的学生。")

    def show_all_students(self):
        if len(self.students) == 0:
            print("学生信息为空。")
        else:
            print("所有学生信息如下：")
            for student in self.students:
                print(f"ID: {student.id}, 姓名: {student.name}, 年龄: {student.age}, 性别: {student.gender}")

def main():
    sms = StudentManagementSystem()
    while True:
        print("""
        1. 添加学生
        2. 删除学生
        3. 更新学生信息
        4. 查找学生
        5. 显示所有学生
        6. 退出
        """)
        choice = input("请选择操作：")
        if choice == '1':
            sms.add_student()
        elif choice == '2':
            sms.delete_student()
        elif choice == '3':
            sms.update_student()
        elif choice == '4':
            sms.search_student()
        elif choice == '5':
            sms.show_all_students()
        elif choice == '6':
            print("退出程序。")
            break
        else:
            print("无效操作。")

if __name__ == '__main__':
    main()