Разработка алгоритма для автоматического составления расписания

def schedule(classes, rooms, timeslots):
    from ortools.sat.python import cp_model
    model = cp_model.CpModel()
 
    # Создание переменных
    class_assignments = {}
    for class_id in classes:
       for room_id in rooms:
            for slot in timeslots:
              class_assignments[(class_id, room_id, slot)] = model.NewBoolVar(f'ca_{class_id}_{room_id}_{slot}')
 
    # Ограничения
    for class_id in classes:
        model.Add(sum(class_assignments[(class_id, room, slot)] for room in rooms for slot in timeslots) == 1)

    for slot in timeslots:
        for room in rooms:
            model.Add(sum(class_assignments[(class_id, room, slot)] for class_id in classes) <= 1)
 
    # Решение
    solver = cp_model.CpSolver()
    status = solver.Solve(model)
    if status == cp_model.OPTIMAL:
        return [(class_id, room, slot) for class_id in classes for room in rooms for slot in timeslots if solver.Value(class_assignments[(class_id, room, slot)]) == 1]
    return []

# Пример данных
classes = ['Math', 'Science']
rooms = ['Room1', 'Room2']
timeslots = ['Morning', 'Afternoon']

# Вызов функции
schedule_result = schedule(classes, rooms, timeslots)