69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any, Dict, Optional
|
|
from fastapi import APIRouter, Depends, HTTPException, Query, Request
|
|
|
|
from app.engine import DuckDBEngine
|
|
|
|
router = APIRouter(tags=["Dashboards"])
|
|
|
|
|
|
def get_engine(request: Request) -> DuckDBEngine:
|
|
return request.app.state.engine
|
|
|
|
|
|
@router.get("/mila/dashboard")
|
|
async def get_mila_dashboard(
|
|
as_of: Optional[str] = Query(default=None, description="Optional target date YYYY-MM-DD"),
|
|
engine: DuckDBEngine = Depends(get_engine),
|
|
) -> Dict[str, Any]:
|
|
try:
|
|
return engine.get_mila_dashboard(as_of_date=as_of)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@router.get("/moritz/dashboard")
|
|
async def get_moritz_dashboard(
|
|
as_of: Optional[str] = Query(default=None, description="Optional target date YYYY-MM-DD"),
|
|
engine: DuckDBEngine = Depends(get_engine),
|
|
) -> Dict[str, Any]:
|
|
try:
|
|
return engine.get_moritz_dashboard(as_of_date=as_of)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@router.get("/mila/day/{date}")
|
|
async def get_mila_day_detail(
|
|
date: str,
|
|
engine: DuckDBEngine = Depends(get_engine),
|
|
) -> Dict[str, Any]:
|
|
try:
|
|
return engine.get_mila_day_detail(target_date=date)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@router.get("/moritz/day/{date}")
|
|
async def get_moritz_day_detail(
|
|
date: str,
|
|
engine: DuckDBEngine = Depends(get_engine),
|
|
) -> Dict[str, Any]:
|
|
try:
|
|
return engine.get_moritz_day_detail(target_date=date)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@router.get("/summary/{subject}")
|
|
async def get_summary(
|
|
subject: str,
|
|
date: Optional[str] = Query(default=None, description="Target date YYYY-MM-DD"),
|
|
engine: DuckDBEngine = Depends(get_engine),
|
|
) -> Dict[str, Any]:
|
|
try:
|
|
return engine.get_summary(subject=subject, date_str=date)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|