From 89168356544da486bdab7485caca724bacd14327 Mon Sep 17 00:00:00 2001
From: Maximilian Giller <m.giller.dev@gmail.com>
Date: Thu, 2 May 2024 17:48:37 +0200
Subject: [PATCH] Added calibration endpoints

---
 src/endpoints/bettwaage.py | 41 ++++++++++++++++++++++++++++++++++++--
 src/main.py                |  2 +-
 2 files changed, 40 insertions(+), 3 deletions(-)

diff --git a/src/endpoints/bettwaage.py b/src/endpoints/bettwaage.py
index 02781f5..9319eb1 100644
--- a/src/endpoints/bettwaage.py
+++ b/src/endpoints/bettwaage.py
@@ -1,7 +1,7 @@
 from fastapi import APIRouter
 
 from datetime import datetime
-from os.path import exists
+import os
 
 from fastapi.responses import HTMLResponse
 
@@ -9,12 +9,20 @@ router = APIRouter()
 
 file_path = "bettwaage.csv"
 
+latest_values = None
+zero_values = [0, 0, 0, 0]
+scale_values = [1, 1, 1, 1]
+
 
 def add_line_to_history(line: str) -> None:
     with open(file_path, "+a", encoding="UTF-8") as fp:
         fp.write(line)
 
 
+def convert_to_weight(value: int, zero_value: int, scale: float) -> float:
+    return (value - zero_value) * scale
+
+
 @router.get("/file")
 async def get_file():
     with open(file_path, "r", encoding="UTF-8") as fp:
@@ -23,9 +31,38 @@ async def get_file():
 
 @router.post("/add")
 async def add_weight(tl: int, tr: int, bl: int, br: int):
+    global latest_values
+    latest_values = [tl, tr, bl, br]
+
+    tl = convert_to_weight(tl, zero_values[0], scale_values[0])
+    tr = convert_to_weight(tl, zero_values[1], scale_values[1])
+    bl = convert_to_weight(tl, zero_values[2], scale_values[2])
+    br = convert_to_weight(tl, zero_values[3], scale_values[3])
+
     sum = tl + tr + bl + br
     add_line_to_history(f"{str(datetime.now())};{tl};{tr};{bl};{br};{sum};")
 
 
-if not exists(file_path):
+@router.post("/delete")
+async def delete_file():
+    os.remove(file_path)
+
+
+@router.post("/zero")
+async def set_zero():
+    if latest_values is None:
+        return HTMLResponse(
+            status_code=400, content="Requiring data before setting zeros."
+        )
+    global zero_values
+    zero_values = latest_values
+
+
+@router.post("/scales")
+async def set_scales(tl: float, tr: float, bl: float, br: float):
+    global scale_values
+    scale_values = [tl, tr, bl, br]
+
+
+if not os.path.exists(file_path):
     add_line_to_history("timestamp;tl;tr;bl;br;total;")
diff --git a/src/main.py b/src/main.py
index 39c4267..28fcb83 100644
--- a/src/main.py
+++ b/src/main.py
@@ -5,7 +5,7 @@ from endpoints.bettwaage import router as bettwaage_router
 app = FastAPI()
 
 app.include_router(hue_router, prefix="/hue", tags=["hue"])
-app.include_router(bettwaage_router, prefix="/bettwaage")
+app.include_router(bettwaage_router, prefix="/bettwaage", tags=["bett"])
 
 if __name__ == "__main__":
     app.run()