Updated for HA 2022.2.6 compatibilty

device_state_attributes replaced with extra_state_attributes

Author: jlvaillant

custom_components/intellicenter/__init__.py

@@ -224,7 +224,7 @@ def __init__(
         self._controller = controller
         self._poolObject = poolObject
         self._available = True
-        self._extraStateAttributes = extraStateAttributes
+        self._extra_state_attributes = extraStateAttributes
         self._name = name
         self._attribute_key = attribute_key
         self._enabled_by_default = enabled_by_default
@@ -314,7 +314,7 @@ def device_info(self):
         }
 
     @property
-    def device_state_attributes(self) -> Optional[Dict[str, Any]]:
+    def extra_state_attributes(self) -> Optional[Dict[str, Any]]:
         """Return the state attributes of the entity."""
 
         object = self._poolObject
@@ -328,7 +328,7 @@ def device_state_attributes(self) -> Optional[Dict[str, Any]]:
         if object.status:
             attributes["Status"] = object.status
 
-        for attribute in self._extraStateAttributes:
+        for attribute in self._extra_state_attributes:
             if object[attribute]:
                 attributes[attribute] = object[attribute]
 

custom_components/intellicenter/light.py

@@ -91,7 +91,7 @@ def __init__(
         """Initialize."""
         super().__init__(entry, controller, poolObject)
         # USE appears to contain extra info like color...
-        self._extraStateAttributes = [USE_ATTR]
+        self._extra_state_attributes = [USE_ATTR]
 
         self._features = 0
 

custom_components/intellicenter/switch.py

@@ -107,7 +107,7 @@ class PoolBody(PoolCircuit):
     def __init__(self, entry: ConfigEntry, controller, poolObject):
         """Initialize a Pool body from the underlying circuit."""
         super().__init__(entry, controller, poolObject)
-        self._extraStateAttributes = [VOL_ATTR, HEATER_ATTR, HTMODE_ATTR]
+        self._extra_state_attributes = [VOL_ATTR, HEATER_ATTR, HTMODE_ATTR]
 
     @property
     def icon(self):

custom_components/intellicenter/water_heater.py

@@ -94,10 +94,10 @@ def __init__(
         self._lastHeater = self._poolObject[HEATER_ATTR]
 
     @property
-    def device_state_attributes(self) -> Optional[Dict[str, Any]]:
+    def extra_state_attributes(self) -> Optional[Dict[str, Any]]:
         """Return the state attributes of the entity."""
 
-        state_attributes = super().device_state_attributes
+        state_attributes = super().extra_state_attributes
 
         if self._lastHeater != NULL_OBJNAM:
             state_attributes[self.LAST_HEATER_ATTR] = self._lastHeater