Ensure this only applies to persistent connections

custom_components/bluetti_bt/binary_sensor.py

@@ -105,8 +105,8 @@ class BluettiBinarySensor(CoordinatorEntity, BinarySensorEntity):
     def _handle_coordinator_update(self) -> None:
         """Handle updated data from the coordinator."""
 
-        if not self.coordinator.client.is_connected:
-          return
+        if self.coordinator.persistent_conn and not self.coordinator.client.is_connected:
+            return
 
         _LOGGER.debug("Updating state of %s", unique_id_loggable(self._attr_unique_id))
         if not isinstance(self.coordinator.data, dict):

custom_components/bluetti_bt/sensor.py

@@ -131,7 +131,7 @@ class BluettiSensor(CoordinatorEntity, SensorEntity):
     def _handle_coordinator_update(self) -> None:
         """Handle updated data from the coordinator."""
 
-        if not self.coordinator.client.is_connected:
+        if self.coordinator.persistent_conn and not self.coordinator.client.is_connected:
             return
 
         _LOGGER.debug("Updating state of %s", unique_id_loggable(self._attr_unique_id))

custom_components/bluetti_bt/switch.py

@@ -115,7 +115,7 @@ class BluettiSwitch(CoordinatorEntity, SwitchEntity):
     def _handle_coordinator_update(self) -> None:
         """Handle updated data from the coordinator."""
 
-        if not self.coordinator.client.is_connected:
+        if self.coordinator.persistent_conn and not self.coordinator.client.is_connected:
             return
 
         _LOGGER.debug("Updating state of %s", unique_id_loggable(self._attr_unique_id))