frontend

backend/app.py

@@ -1,9 +1,11 @@
+from contextlib import asynccontextmanager
 from pathlib import Path
 from typing import Any
 from datetime import date, datetime
 import os
 
 from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel, Field
 
 import db
@@ -15,9 +17,7 @@ from scheduler import SchedulerTest, compile_schedule, remove_from_active, reset
 APP_ROOT = Path(__file__).resolve().parent
 DB_PATH = APP_ROOT / "scheduler.db"
 DUT = os.getenv("DUT", "CGW453").strip()
-REF = os.getenv("REF", "CGW452").strip() 
-
-app = FastAPI(title="Scheduler API", version="0.1.0")
+REF = os.getenv("REF", "CGW452").strip()
 
 
 class LoadTestsRequest(BaseModel):
@@ -40,10 +40,25 @@ class RemoveActiveTestsRequest(BaseModel):
 	test_ids: list[str] = Field(default_factory=list)
 
 
-@app.on_event("startup")
-def on_startup() -> None:
+class SaveHolidaysRequest(BaseModel):
+	dates: list[str] = Field(default_factory=list)
+
+
+@asynccontextmanager
+async def lifespan(application: FastAPI):
 	db.init_db(DB_PATH)
 	graph.reset_graph_state()
+	yield
+
+
+app = FastAPI(title="Scheduler API", version="0.1.0", lifespan=lifespan)
+
+app.add_middleware(
+	CORSMiddleware,
+	allow_origins=["http://localhost:5173", "http://127.0.0.1:5173"],
+	allow_methods=["*"],
+	allow_headers=["*"],
+)
 
 
 @app.get("/api/health")
@@ -154,6 +169,18 @@ def compile_schedule_endpoint(request: CompileScheduleRequest) -> dict[str, Any]
 	}
 
 
+@app.post("/api/holidays")
+def save_holidays(request: SaveHolidaysRequest) -> dict[str, Any]:
+	dates = [d.strip() for d in request.dates if d.strip()]
+	db.upsert_holidays(dates, DB_PATH)
+	return {"status": "saved", "count": len(dates)}
+
+
+@app.get("/api/holidays")
+def get_holidays() -> dict[str, Any]:
+	return {"dates": sorted(db.list_holidays(DB_PATH))}
+
+
 @app.get("/api/schedule/week")
 def get_schedule_week(start: str | None = None) -> dict[str, Any]:
 	week_start = start or date.today().isoformat()
@@ -174,6 +201,7 @@ def get_schedule_week(start: str | None = None) -> dict[str, Any]:
 				"sequence_in_shift": row.sequence_in_shift,
 				"test_type": row.test_type,
 				"rotation": row.rotation,
+				"config": row.config,
 				"status": row.status,
 				"priority": row.priority,
 				"estimated_minutes": row.estimated_minutes,
@@ -181,3 +209,8 @@ def get_schedule_week(start: str | None = None) -> dict[str, Any]:
 			for row in rows
 		],
 	}
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run("app:app", host="0.0.0.0", port=8000, reload=True)
+