Added availability status

klejejs · Jan 4, 2022 · dd01ce4 · dd01ce4
1 parent 6aa29cf
commit dd01ce4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/custom_components/thermia/sensor.py b/custom_components/thermia/sensor.py
@@ -38,6 +38,11 @@ def __init__(self, coordinator, idx):
         super().__init__(coordinator)
         self.idx = idx
 
+    @property
+    def available(self):
+        """Return True if entity is available."""
+        return self.coordinator.data.heat_pumps[self.idx].is_online
+
     @property
     def name(self):
         """Return the name of the sensor."""

diff --git a/custom_components/thermia/water_heater.py b/custom_components/thermia/water_heater.py
@@ -49,6 +49,11 @@ def __init__(self, coordinator, idx):
         super().__init__(coordinator)
         self.idx = idx
 
+    @property
+    def available(self):
+        """Return True if entity is available."""
+        return self.coordinator.data.heat_pumps[self.idx].is_online
+
     @property
     def name(self):
         """Return the name of the water heater."""