Restore device_sn property

jjjonesjr33 · Dec 24, 2024 · 2aa8d3a · 2aa8d3a
1 parent 438c764
commit 2aa8d3a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/custom_components/petlibro/devices/feeders/polar_wet_food_feeder.py b/custom_components/petlibro/devices/feeders/polar_wet_food_feeder.py
@@ -35,6 +35,11 @@ async def refresh(self):
     def battery_state(self) -> str | None:
         return self._data.get("batteryState")
 
+    @property
+    def device_sn(self) -> str:
+        """Returns the serial number of the device."""
+        return self._data.get("deviceSn", "unknown")
+
     @property
     def door_blocked(self) -> bool | None:
         return self._data.get("realInfo", {}).get("barnDoorError")

diff --git a/custom_components/petlibro/sensor.py b/custom_components/petlibro/sensor.py
@@ -543,7 +543,7 @@ def device_class(self) -> SensorDeviceClass | None:
     ],
     PolarWetFoodFeeder: [
         PetLibroSensorEntityDescription[PolarWetFoodFeeder](
-            key="serial",
+            key="device_sn",
             translation_key="device_sn",
             icon="mdi:identifier",
             name="Device SN"