from typing import Literal, TypedDict, List Priority = Literal["low", "normal", "high", "critical"] class Task(TypedDict): task_id: str task_type: str priority: Priority source: str # user, schedule, sensor, system class ScheduledTask(TypedDict): task_id: str order: int reason: str def schedule_tasks(tasks: List[Task]) -> List[ScheduledTask]: """Order tasks using simple priority and source rules. - critical > high > normal > low - user > sensor > schedule > system (within same priority) """ pr_order = {"critical": 0, "high": 1, "normal": 2, "low": 3} src_order = {"user": 0, "sensor": 1, "schedule": 2, "system": 3} sorted_tasks = sorted( tasks, key=lambda t: ( pr_order.get(t["priority"], 2), src_order.get(t["source"], 3), ), ) result: List[ScheduledTask] = [] for idx, t in enumerate(sorted_tasks, start=1): reason = f"priority={t['priority']}, source={t['source']} -> position {idx}." result.append({"task_id": t["task_id"], "order": idx, "reason": reason}) return result if __name__ == "__main__": sample: List[Task] = [ {"task_id": "T_CLEAN_KITCHEN", "task_type": "clean_room", "priority": "high", "source": "schedule"}, {"task_id": "T_REMIND_TRASH", "task_type": "reminder", "priority": "high", "source": "sensor"}, {"task_id": "T_PATROL_NIGHT", "task_type": "patrol", "priority": "normal", "source": "schedule"}, {"task_id": "T_STATUS", "task_type": "status_report", "priority": "low", "source": "system"}, {"task_id": "T_CLEAN_LIVING", "task_type": "clean_room", "priority": "critical", "source": "user"}, ] for st in schedule_tasks(sample): print(st)