Climate and sensor report as not available when grill is offline

sebirdman · Feb 13, 2021 · 24cc1fb · 24cc1fb
1 parent f5eaba2
commit 24cc1fb
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/custom_components/traeger/climate.py b/custom_components/traeger/climate.py
@@ -67,6 +67,14 @@ def grill_update(self):
         self.schedule_update_ha_state()
 
     # Generic Properties
+    @property
+    def available(self):
+        """Reports unavailable when the grill is powered off"""
+        if self.grill_state is None:
+            return False
+        else:
+            return True if self.grill_state["connected"] == True else False
+
     @property
     def name(self):
         """Return the name of the grill"""

diff --git a/custom_components/traeger/sensor.py b/custom_components/traeger/sensor.py
@@ -54,6 +54,14 @@ def grill_update(self):
         self.schedule_update_ha_state()
 
     # Generic Properties
+    @property
+    def available(self):
+        """Reports unavailable when the grill is powered off"""
+        if self.grill_state is None:
+            return False
+        else:
+            return True if self.grill_state["connected"] == True else False
+
     @property
     def name(self):
         """Return the name of the sensor."""
@@ -106,6 +114,14 @@ def grill_update(self):
         self.schedule_update_ha_state()
 
     # Generic Properties
+    @property
+    def available(self):
+        """Reports unavailable when the grill is powered off"""
+        if self.grill_state is None:
+            return False
+        else:
+            return True if self.grill_state["connected"] == True else False
+
     @property
     def name(self):
         """Return the name of the sensor."""