77 lines
2.8 KiB
Python
77 lines
2.8 KiB
Python
import json
|
|
import logging
|
|
|
|
from fastapi import APIRouter, Depends, Request, status
|
|
from fastapi.responses import PlainTextResponse, Response
|
|
from pydantic import ValidationError
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.config import Settings
|
|
from app.dependencies import get_app_settings, get_homeassistant_client, get_poo_db
|
|
from app.integrations.homeassistant import HomeAssistantClient
|
|
from app.schemas.poo import PooRecordRequest
|
|
from app.services.poo import publish_latest_poo_status, record_poo
|
|
|
|
router = APIRouter(tags=["poo"])
|
|
logger = logging.getLogger(__name__)
|
|
BAD_REQUEST_MESSAGE = "bad request"
|
|
INTERNAL_SERVER_ERROR_MESSAGE = "internal server error"
|
|
|
|
|
|
@router.post("/poo/record")
|
|
async def create_poo_record(
|
|
request: Request,
|
|
db: Session = Depends(get_poo_db),
|
|
settings: Settings = Depends(get_app_settings),
|
|
homeassistant_client: HomeAssistantClient = Depends(get_homeassistant_client),
|
|
) -> Response:
|
|
try:
|
|
raw_payload = await request.body()
|
|
data = json.loads(raw_payload)
|
|
payload = PooRecordRequest.model_validate(data)
|
|
record_poo(
|
|
db,
|
|
payload,
|
|
settings=settings,
|
|
homeassistant_client=homeassistant_client,
|
|
)
|
|
except json.JSONDecodeError as exc:
|
|
logger.warning("Rejected poo record request due to invalid JSON: %s", exc)
|
|
return PlainTextResponse(BAD_REQUEST_MESSAGE, status_code=status.HTTP_400_BAD_REQUEST)
|
|
except ValidationError as exc:
|
|
logger.warning("Rejected poo record request due to validation failure: %s", exc)
|
|
return PlainTextResponse(BAD_REQUEST_MESSAGE, status_code=status.HTTP_400_BAD_REQUEST)
|
|
except ValueError as exc:
|
|
logger.warning("Rejected poo record request due to invalid numeric input: %s", exc)
|
|
return PlainTextResponse(BAD_REQUEST_MESSAGE, status_code=status.HTTP_400_BAD_REQUEST)
|
|
except Exception as exc:
|
|
logger.warning("Failed to store poo record: %s", exc)
|
|
return PlainTextResponse(
|
|
INTERNAL_SERVER_ERROR_MESSAGE,
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
)
|
|
|
|
return Response(status_code=status.HTTP_200_OK)
|
|
|
|
|
|
@router.get("/poo/latest")
|
|
def notify_latest_poo(
|
|
db: Session = Depends(get_poo_db),
|
|
settings: Settings = Depends(get_app_settings),
|
|
homeassistant_client: HomeAssistantClient = Depends(get_homeassistant_client),
|
|
) -> Response:
|
|
try:
|
|
publish_latest_poo_status(
|
|
session=db,
|
|
settings=settings,
|
|
homeassistant_client=homeassistant_client,
|
|
)
|
|
except Exception as exc:
|
|
logger.warning("Failed to publish latest poo status: %s", exc)
|
|
return PlainTextResponse(
|
|
INTERNAL_SERVER_ERROR_MESSAGE,
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
)
|
|
|
|
return Response(status_code=status.HTTP_200_OK)
|