Set unit of unavailable data if unit is available in api response

Close #5
m-rossi · Oct 14, 2024 · 71a860c · 71a860c
1 parent d25f012
commit 71a860c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/custom_components/dashboard_delbrueck/sensor.py b/custom_components/dashboard_delbrueck/sensor.py
@@ -75,6 +75,8 @@ def update(self):
                 self._attr_native_unit_of_measurement = UNIT_CONVERTER[data["unit"]]
         except (ValueError, ConnectionError):
             self._attr_native_value = None
+            if "unit" in data and data["unit"] in UNIT_CONVERTER:
+                self._attr_native_unit_of_measurement = UNIT_CONVERTER[data["unit"]]
 
 
 def replace_umlauts(text: str) -> str: