add homekeeper stack

api/app/routers/apiaries.py

@@ -0,0 +1,99 @@
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy import text
+from sqlalchemy.orm import Session
+from typing import Any
+
+from app.database import get_session
+from app.auth import get_current_user
+
+router = APIRouter()
+
+
+def _row(r) -> dict:
+    return dict(r._mapping)
+
+
+@router.get("")
+def list_apiaries(
+    session: Session = Depends(get_session),
+    _: str = Depends(get_current_user),
+) -> list[dict]:
+    rows = session.execute(text("""
+        SELECT a.*, COUNT(c.id) AS colony_count
+        FROM apiaries a
+        LEFT JOIN colonies c ON c.apiary_id = a.id AND c.closed_at IS NULL
+        GROUP BY a.id
+        ORDER BY a.name
+    """)).fetchall()
+    return [_row(r) for r in rows]
+
+
+@router.get("/{id}")
+def get_apiary(
+    id: int,
+    session: Session = Depends(get_session),
+    _: str = Depends(get_current_user),
+) -> dict:
+    row = session.execute(
+        text("SELECT * FROM apiaries WHERE id = :id"), {"id": id}
+    ).fetchone()
+    if row is None:
+        raise HTTPException(status_code=404, detail="Not found")
+    return _row(row)
+
+
+@router.post("", status_code=201)
+def create_apiary(
+    body: dict[str, Any],
+    session: Session = Depends(get_session),
+    _: str = Depends(get_current_user),
+) -> dict:
+    row = session.execute(text("""
+        INSERT INTO apiaries (name, location_text, lat, lng, notes)
+        VALUES (:name, :location_text, :lat, :lng, :notes)
+        RETURNING *
+    """), {
+        "name": body.get("name"),
+        "location_text": body.get("location_text"),
+        "lat": body.get("lat"),
+        "lng": body.get("lng"),
+        "notes": body.get("notes"),
+    }).fetchone()
+    session.commit()
+    return _row(row)
+
+
+@router.put("/{id}")
+def update_apiary(
+    id: int,
+    body: dict[str, Any],
+    session: Session = Depends(get_session),
+    _: str = Depends(get_current_user),
+) -> dict:
+    result = session.execute(text("""
+        UPDATE apiaries
+        SET name = :name, location_text = :location_text, lat = :lat, lng = :lng, notes = :notes
+        WHERE id = :id
+        RETURNING *
+    """), {
+        "id": id,
+        "name": body.get("name"),
+        "location_text": body.get("location_text"),
+        "lat": body.get("lat"),
+        "lng": body.get("lng"),
+        "notes": body.get("notes"),
+    }).fetchone()
+    session.commit()
+    if result is None:
+        raise HTTPException(status_code=404, detail="Not found")
+    return _row(result)
+
+
+@router.delete("/{id}", status_code=204)
+def delete_apiary(
+    id: int,
+    session: Session = Depends(get_session),
+    _: str = Depends(get_current_user),
+):
+    session.execute(text("DELETE FROM apiaries WHERE id = :id"), {"id": id})
+    session.commit()