feat(#170): add calibrate weight button and weight calibration error sensor for DockstreamSmartFountain (#254)

Co-authored-by: Jamie Jones <29973406+jjjonesjr33@users.noreply.github.com>

Author: dennismdejong

custom_components/petlibro/binary_sensor.py

@@ -629,6 +629,14 @@ def extra_state_attributes(self):
             should_report=lambda device: device.light_switch is not None,
             name="Indicator"
         ),
+        PetLibroBinarySensorEntityDescription[DockstreamSmartFountain](
+            key="weight_calibration_error",
+            translation_key="weight_calibration_error",
+            icon="mdi:alert",
+            device_class=BinarySensorDeviceClass.PROBLEM,
+            should_report=lambda device: device.weight_calibration_error is not None,
+            name="Weight Calibration Error"
+        ),
     ],
     DockstreamSmartRFIDFountain: [
         PetLibroBinarySensorEntityDescription[DockstreamSmartRFIDFountain](

custom_components/petlibro/button.py

@@ -657,7 +657,14 @@ class PetLibroButtonEntityDescription(ButtonEntityDescription, PetLibroEntityDes
             translation_key="filter_reset",
             set_fn=lambda device: device.set_filter_reset(),
             name="Filter Reset"
-        )
+        ),
+        PetLibroButtonEntityDescription[DockstreamSmartFountain](
+            key="calibrate_weight",
+            translation_key="calibrate_weight",
+            icon="mdi:scale",
+            set_fn=lambda device: device.calibrate_weight(),
+            name="Calibrate Weight Sensor"
+        ),
     ],
     DockstreamSmartRFIDFountain: [
         PetLibroButtonEntityDescription[DockstreamSmartRFIDFountain](

custom_components/petlibro/devices/fountains/dockstream_smart_fountain.py

@@ -6,7 +6,6 @@
 from ..device import Device
 from typing import cast
 from logging import getLogger
-
 _LOGGER = getLogger(__name__)
 
 class DockstreamSmartFountain(Device):
@@ -356,3 +355,19 @@ def update_progress(self) -> float:
 
         progress = upgrade_data.get("progress")
         return float(progress) if progress is not None else 0.0
+
+    @property
+    def weight_calibration_error(self) -> bool | None:
+        exception = self._data.get("dataRealInfo", {}).get("exceptionMessage")
+        if exception is None:
+            return None
+        return "calibration" in exception.lower() or "weight" in exception.lower()
+
+    async def calibrate_weight(self) -> None:
+        _LOGGER.debug(f"Calibrating weight sensor for {self.serial}")
+        try:
+            await self.api.exec_device_command(self.serial, "CALIBRATE")
+            await self.refresh()
+        except (aiohttp.ClientError, PetLibroAPIError) as err:
+            _LOGGER.error(f"Failed to calibrate weight sensor for {self.serial}: {err}")
+            raise PetLibroAPIError(f"Error calibrating weight sensor: {err}")