Set enabled flag for wifi signal sensor

rospogrigio · Oct 28, 2022 · 7ae1d1e · 7ae1d1e
1 parent 127e2e2
commit 7ae1d1e
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/custom_components/daikin_residential/sensor.py b/custom_components/daikin_residential/sensor.py
@@ -203,12 +203,9 @@ def state_class(self):
 
 class DaikinGatewaySensor(DaikinSensor):
     """Representation of a WiFi Sensor."""
-    
+
     # set default category for these entities
     _attr_entity_category = EntityCategory.DIAGNOSTIC
-
-    # auto disable these entities when added for the first time
-    _attr_entity_registry_enabled_default = False
 
     @property
     def state(self):
@@ -231,3 +228,8 @@ def state_class(self):
             return STATE_CLASS_MEASUREMENT
         else:
             return None
+
+    @property
+    def entity_registry_enabled_default(self):
+        # auto disable these entities when added for the first time except the wifi signal
+        return (self._device_attribute == ATTR_WIFI_STRENGTH)