Create main.py

main.py

@@ -0,0 +1,47 @@
+# Additional imports
+from typing import List, Dict
+
+# Merge roadmaps
+def merge_roadmaps(regular_roadmap: Dict, test_roadmap: Dict) -> Dict:
+    """Merge two roadmaps into one, combining overlapping and non-overlapping dates."""
+    def schedule_to_dict(schedule: List[Dict]) -> Dict:
+        """Convert a roadmap's schedule into a dictionary indexed by dates."""
+        return {day["date"]: day for day in schedule}
+
+    # Convert schedules to dictionaries for quick lookup
+    regular_schedule_dict = schedule_to_dict(regular_roadmap["schedule"])
+    test_schedule_dict = schedule_to_dict(test_roadmap["schedule"])
+
+    # Parse dates from schedules
+    regular_dates = [datetime.strptime(day["date"], "%Y-%m-%d") for day in regular_roadmap["schedule"]]
+    test_dates = [datetime.strptime(day["date"], "%Y-%m-%d") for day in test_roadmap["schedule"]]
+
+    # Identify start and end dates for both roadmaps
+    regular_start, regular_end = min(regular_dates), max(regular_dates)
+    test_start, test_end = min(test_dates), max(test_dates)
+
+    # Merge overlapping dates
+    for date, test_day in test_schedule_dict.items():
+        if date in regular_schedule_dict:
+            # Append test roadmap subjects to matching days in regular roadmap
+            regular_schedule_dict[date]["subjects"].extend(test_day["subjects"])
+        else:
+            # Add non-overlapping test roadmap dates
+            regular_schedule_dict[date] = test_day
+
+    # Convert the merged schedule dictionary back to a sorted list
+    merged_schedule = sorted(regular_schedule_dict.values(), key=lambda x: x["date"])
+
+    return {"schedule": merged_schedule}
+
+# New endpoint for merging roadmaps
+@app.post("/merge-roadmaps/")
+def merge_roadmaps_endpoint(roadmap1: Dict, roadmap2: Dict):
+    """
+    Merge two roadmaps into one, combining overlapping and non-overlapping dates.
+    """
+    try:
+        merged_roadmap = merge_roadmaps(roadmap1, roadmap2)
+        return merged_roadmap
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error merging roadmaps: {e}")