From bc365ffe7a2e24c79b331c1b73bf3f864fd70f27 Mon Sep 17 00:00:00 2001 From: Tianyu Liu Date: Wed, 7 Aug 2024 14:44:24 +0200 Subject: [PATCH] Change poo record to post --- src/main.py | 14 ++++++++++---- src/recorder/poo.py | 2 +- 2 files changed, 11 insertions(+), 5 deletions(-) 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", },