Add location in poo record

This commit is contained in:
2024-08-24 22:11:15 +02:00
parent b7aaca1650
commit 9a5fc701bb
2 changed files with 15 additions and 13 deletions

View File

@@ -1,7 +1,6 @@
from contextlib import asynccontextmanager
from fastapi import FastAPI
from pydantic import BaseModel
from src.cloud_util.homeassistant import HomeAssistant
from src.cloud_util.mqtt import MQTT
@@ -26,10 +25,6 @@ async def _lifespan(_app: FastAPI): # noqa: ANN202
await mqtt.stop()
class PooRecordField(BaseModel):
status: str
app = FastAPI(lifespan=_lifespan)
@@ -45,8 +40,8 @@ async def homeassistant_publish(payload: HomeAssistant.Message) -> dict:
# Poo recorder
@app.post("/poo/record")
async def record(record_detail: PooRecordField) -> PooRecordField:
await poo_recorder.record(record_detail.status)
async def record(record_detail: PooRecorder.RecordField) -> PooRecorder.RecordField:
await poo_recorder.record(record_detail)
return record_detail