fixed entity update

tomaae · May 10, 2023 · b9fe492 · b9fe492
1 parent 4b30177
commit b9fe492
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/custom_components/portainer/entity.py b/custom_components/portainer/entity.py
@@ -105,7 +105,14 @@ def __init__(
         self._uid = uid
         self._data = coordinator.data[self.description.data_path]
         if self._uid:
-            self._data = self._data[self._uid]
+            self._data = coordinator.data[self.description.data_path][self._uid]
+
+    @callback
+    def _handle_coordinator_update(self) -> None:
+        self._data = self.coordinator.data[self.description.data_path]
+        if self._uid:
+            self._data = self.coordinator.data[self.description.data_path][self._uid]
+        super()._handle_coordinator_update()
 
     @property
     def name(self) -> str: