Initial commit: Waste Collection API

app/routes/collections.py

@@ -0,0 +1,56 @@
+from fastapi import APIRouter, HTTPException
+
+from ..schemas import (
+    ICSRequest,
+    AbfallIORequest,
+    AWMRequest,
+    StuttgartRequest,
+    CollectionsResponse,
+)
+from ..collector import fetch_ics, fetch_abfall_io, fetch_awm_muenchen, fetch_stuttgart
+
+router = APIRouter(prefix="/collections", tags=["collections"])
+
+
+@router.post("/ics", response_model=CollectionsResponse)
+def collections_ics(body: ICSRequest):
+    try:
+        collections = fetch_ics(body.url, body.count)
+    except Exception as e:
+        raise HTTPException(status_code=422, detail=str(e))
+    if not collections:
+        raise HTTPException(status_code=422, detail="No collections returned from this ICS URL")
+    return CollectionsResponse(source="ics", collections=collections)
+
+
+@router.post("/abfall_io", response_model=CollectionsResponse)
+def collections_abfall_io(body: AbfallIORequest):
+    try:
+        collections = fetch_abfall_io(body.key, body.f_id_kommune, body.f_id_strasse, body.count)
+    except Exception as e:
+        raise HTTPException(status_code=422, detail=str(e))
+    if not collections:
+        raise HTTPException(status_code=422, detail="No collections returned. Check your IDs.")
+    return CollectionsResponse(source="abfall_io", collections=collections)
+
+
+@router.post("/awm_muenchen_de", response_model=CollectionsResponse)
+def collections_awm_muenchen(body: AWMRequest):
+    try:
+        collections = fetch_awm_muenchen(body.street, body.house_number, body.count)
+    except Exception as e:
+        raise HTTPException(status_code=422, detail=str(e))
+    if not collections:
+        raise HTTPException(status_code=422, detail="No collections returned. Check street name.")
+    return CollectionsResponse(source="awm_muenchen_de", collections=collections)
+
+
+@router.post("/stuttgart_de", response_model=CollectionsResponse)
+def collections_stuttgart(body: StuttgartRequest):
+    try:
+        collections = fetch_stuttgart(body.street, body.streetnr, body.count)
+    except Exception as e:
+        raise HTTPException(status_code=422, detail=str(e))
+    if not collections:
+        raise HTTPException(status_code=422, detail="No collections returned. Check street name and number.")
+    return CollectionsResponse(source="stuttgart_de", collections=collections)