Create timetable.py

timetable.py

@@ -0,0 +1,56 @@
+import pandas as pd
+from datetime import time
+
+DAYS = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday"]
+
+def generate_time_slots():
+    slots = []
+    start_hour = 9.0
+    end_hour = 15.0
+
+    while start_hour < end_hour:
+        slots.append(start_hour)
+        start_hour += 1.5 if start_hour + 1.5 <= end_hour else 1.0
+
+    return slots
+
+
+def generate_timetable(courses_path, rooms_path):
+    courses = pd.read_csv(courses_path)
+    rooms = pd.read_csv(rooms_path)["Room"].tolist()
+
+    slots = generate_time_slots()
+    timetable = []
+
+    day_idx = 0
+    slot_idx = 0
+    room_idx = 0
+
+    for _, row in courses.iterrows():
+        remaining_hours = row["Hours"]
+
+        while remaining_hours > 0:
+            duration = 1.5 if remaining_hours >= 1.5 else 1.0
+            start = slots[slot_idx]
+            end = start + duration
+
+            timetable.append({
+                "Day": DAYS[day_idx],
+                "Time": f"{start:0.1f} - {end:0.1f}",
+                "Course": row["Course"],
+                "Teacher": row["Teacher"],
+                "Room": rooms[room_idx % len(rooms)]
+            })
+
+            remaining_hours -= duration
+            slot_idx += 1
+            room_idx += 1
+
+            if slot_idx >= len(slots):
+                slot_idx = 0
+                day_idx += 1
+
+            if day_idx >= len(DAYS):
+                return pd.DataFrame(timetable)
+
+    return pd.DataFrame(timetable)