diff --git a/src/main.py b/src/main.py index 09caaba..06ef850 100644 --- a/src/main.py +++ b/src/main.py @@ -1,6 +1,7 @@ from contextlib import asynccontextmanager from fastapi import FastAPI +from pydantic import BaseModel from src.recorder.poo import PooRecorder @@ -14,10 +15,15 @@ async def _lifespan(_app: FastAPI): # noqa: ANN202 await recorder.stop() +class PooRecordField(BaseModel): + status: str + + app = FastAPI(lifespan=_lifespan) -@app.put("/record/s={status}") -async def record(status: str) -> dict: - await recorder.record(status) - return {"status": status} +@app.post("/poo/record") +async def record(record_detail: PooRecordField) -> PooRecordField: + print(record_detail.status) + await recorder.record(record_detail.status) + return record_detail diff --git a/src/recorder/poo.py b/src/recorder/poo.py index 6a1d298..7fc91a6 100644 --- a/src/recorder/poo.py +++ b/src/recorder/poo.py @@ -60,7 +60,7 @@ class PooRecorder: "device": { "name": "Dog Poop Recorder", "model": "poop-recorder-backend", - "sw_version": "1.1", + "sw_version": "1.2", "identifiers": ["poo_recorder"], "manufacturer": "Studio TJ", },