写一个考场安排系统idea代码

以下是一个简单的考场安排系统的Idea代码：

# 假设已经有所有考生和考场信息的数据

# 考场信息包括考场编号、考场容量、座位号等
class ExamRoom:
    def __init__(self, room_id, capacity, seats):
        self.room_id = room_id
        self.capacity = capacity
        self.seats = seats

# 考生信息包括姓名、准考证号、所在考场等
class Examinee:
    def __init__(self, name, exam_id, room_id, seat_id):
        self.name = name
        self.exam_id = exam_id
        self.room_id = room_id
        self.seat_id = seat_id

# 考场安排系统类
class ExamArrangementSystem:
    def __init__(self):
        self.exam_rooms = [] # 所有考场信息
        self.examinees = [] # 所有考生信息

    # 添加考场信息
    def add_exam_room(self, room_id, capacity, seats):
        exam_room = ExamRoom(room_id, capacity, seats)
        self.exam_rooms.append(exam_room)

    # 添加考生信息
    def add_examinee(self, name, exam_id, room_id, seat_id):
        examinee = Examinee(name, exam_id, room_id, seat_id)
        self.examinees.append(examinee)

    # 根据考生准考证号查询考生信息
    def search_examinee_by_exam_id(self, exam_id):
        for examinee in self.examinees:
            if examinee.exam_id == exam_id:
                return examinee
        return None

    # 根据考场编号查询考场信息
    def search_exam_room_by_room_id(self, room_id):
        for exam_room in self.exam_rooms:
            if exam_room.room_id == room_id:
                return exam_room
        return None

    # 根据考场编号和座位号查询座位是否空闲
    def is_seat_available(self, room_id, seat_id):
        exam_room = self.search_exam_room_by_room_id(room_id)
        if exam_room is not None:
            if exam_room.seats[seat_id-1] is None:
                return True
        return False

    # 安排考生座位
    def arrange_seat(self, exam_id, room_id, seat_id):
        examinee = self.search_examinee_by_exam_id(exam_id)
        if examinee is not None:
            exam_room = self.search_exam_room_by_room_id(room_id)
            if exam_room is not None:
                if self.is_seat_available(room_id, seat_id):
                    exam_room.seats[seat_id-1] = examinee
                    examinee.room_id = room_id
                    examinee.seat_id = seat_id
                    print(f"{examinee.name}({examinee.exam_id})已安排在{room_id}考场{seat_id}座位")
                else:
                    print("该座位已被占用，请重新选择座位")
            else:
                print("该考场不存在，请重新输入考场编号")
        else:
            print("该考生不存在，请重新输入准考证号")

# 使用示例
exam_arrangement = ExamArrangementSystem()

# 添加考场信息
exam_arrangement.add_exam_room("A101", 50, [None]*50)
exam_arrangement.add_exam_room("A102", 80, [None]*80)

# 添加考生信息
exam_arrangement.add_examinee("张三", "2022010101", None, None)
exam_arrangement.add_examinee("李四", "2022010102", None, None)

# 安排考生座位
exam_arrangement.arrange_seat("2022010101", "A101", 10)
exam_arrangement.arrange_seat("2022010102", "A102", 20)