Track pump activity

matteius · matteius · commit 075f630b985e · 2025-06-28T09:56:43.000-04:00
diff --git a/opensensor/collection_apis.py b/opensensor/collection_apis.py
@@ -23,6 +23,7 @@
     Lux,
     Moisture,
     Pressure,
+    PumpBoard,
     RelayBoard,
     RelayStatus,
     Temperature,
@@ -118,6 +119,7 @@ def get_device_info_cached(device_id: str):
     "pH": "pH",
     "liquid": "liquid",
     "relays": "relays",
+    "pumps": "pumps",
 }
 
 
@@ -185,6 +187,8 @@ def _record_data_to_ts_collection(
                         doc_to_insert[column_name + "_unit"] = value
                     elif key == "relays":
                         doc_to_insert[column_name] = json.dumps(value)
+                    elif key == "pumps":
+                        doc_to_insert[column_name] = json.dumps(value)
                     else:
                         doc_to_insert[column_name] = str(value)
 
@@ -742,6 +746,12 @@ async def historical_data_route(
     response_model=Page[RelayBoard],
     methods=["GET"],
 )
+router.add_api_route(
+    "/pumps/{device_id}",
+    create_historical_data_route(PumpBoard),
+    response_model=Page[PumpBoard],
+    methods=["GET"],
+)
 
 
 @router.post("/environment/")
diff --git a/opensensor/collections.py b/opensensor/collections.py
@@ -77,6 +77,25 @@ def collection_name(cls):
         return "relays"
 
 
+class PumpStatus(BaseModel):
+    position: int
+    enabled: bool
+    speed: float | None = None
+    duration: float | None = None
+    timestamp: float | None = None
+    description: str | None = None
+
+
+class PumpBoard(TimestampModel):
+    """Pump activity tracking"""
+
+    pumps: List[PumpStatus]
+
+    @classmethod
+    def collection_name(cls):
+        return "pumps"
+
+
 class Moisture(TimestampModel):
     readings: List[float | Decimal | int] | str
 
@@ -99,6 +118,7 @@ class Environment(BaseModel):
     pH: PH | None = None
     liquid: LiquidLevel | None = None
     relays: RelayBoard | None = None
+    pumps: PumpBoard | None = None
 
 
 class VPD(BaseModel):
