Prevent diagnostics from interfering with data update

custom_components/dabpumps/api.py

@@ -518,6 +518,10 @@ async def async_get_diagnostics(self) -> dict[str, Any]:
     async def _async_update_diagnostics(self, timestamp, context: str, request: dict|None, response: dict|None, token: dict|None = None):
         # worker function
         async def _async_worker(self, timestamp, context, request, response, token):
+
+            # Wait one or two seconds to not interfere with the updating of the entities after retrieving the data
+            await asyncio.sleep(2)
+
             item = DabPumpsApiHistoryItem(timestamp, context, request, response, token)
             detail = DabPumpsApiHistoryDetail(timestamp, context, request, response, token)
 

custom_components/dabpumps/binary_sensor.py

@@ -116,11 +116,9 @@ def name(self) -> str:
     @callback
     def _handle_coordinator_update(self) -> None:
         """Handle updated data from the coordinator."""
-        super()._handle_coordinator_update()
-
-        (_, _, status_map) = self._coordinator.data
-
+
         # find the correct device and status corresponding to this sensor
+        (_, _, status_map) = self._coordinator.data
         status = status_map.get(self.object_id)
 
         # Update any attributes

custom_components/dabpumps/number.py

@@ -102,11 +102,9 @@ def name(self) -> str:
     @callback
     def _handle_coordinator_update(self) -> None:
         """Handle updated data from the coordinator."""
-        super()._handle_coordinator_update()
-
-        (_, _, status_map) = self._coordinator.data
-
+
         # find the correct device and status corresponding to this sensor
+        (_, _, status_map) = self._coordinator.data
         status = status_map.get(self.object_id)
 
         # Update any attributes

custom_components/dabpumps/select.py

@@ -103,11 +103,9 @@ def name(self) -> str:
     @callback
     def _handle_coordinator_update(self) -> None:
         """Handle updated data from the coordinator."""
-        super()._handle_coordinator_update()
-
-        (_, _, status_map) = self._coordinator.data
 
         # find the correct device and status corresponding to this sensor
+        (_, _, status_map) = self._coordinator.data
         status = status_map.get(self.object_id)
 
         # Update any attributes
@@ -136,6 +134,7 @@ def _update_attributes(self, status, is_create):
             self._name = status.key
 
             self._attr_options = list(self._dict.values())
+            self._attr_current_option = None
 
             self._attr_entity_category = self.get_entity_category()
 

custom_components/dabpumps/sensor.py

@@ -104,11 +104,9 @@ def name(self) -> str:
     @callback
     def _handle_coordinator_update(self) -> None:
         """Handle updated data from the coordinator."""
-        super()._handle_coordinator_update()
-
-        (_, _, status_map) = self._coordinator.data
-
+
         # find the correct device and status corresponding to this sensor
+        (_, _, status_map) = self._coordinator.data
         status = status_map.get(self.object_id)
 
         # Update any attributes
@@ -147,7 +145,7 @@ def _update_attributes(self, status, is_create):
                 attr_precision = None
                 attr_val = self._get_string(str(status.val)) if status.val!=None else None
                 attr_unit = None
-        
+
         # Process any changes
         changed = False
 

custom_components/dabpumps/switch.py

@@ -111,11 +111,9 @@ def name(self) -> str:
     @callback
     def _handle_coordinator_update(self) -> None:
         """Handle updated data from the coordinator."""
-        super()._handle_coordinator_update()
-
-        (_, _, status_map) = self._coordinator.data
 
         # find the correct device and status corresponding to this sensor
+        (_, _, status_map) = self._coordinator.data
         status = status_map.get(self.object_id)
 
         # Update any attributes