Add location in poo record
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user