From 296c9cbfd357054845428f718cb87f7d5931cde0 Mon Sep 17 00:00:00 2001
From: SupaFlav <52390958+SupaFlav@users.noreply.github.com>
Date: Mon, 28 Aug 2023 15:33:45 -0400
Subject: [PATCH] Ensure this only applies to persistent connections

---
 custom_components/bluetti_bt/binary_sensor.py | 4 ++--
 custom_components/bluetti_bt/sensor.py        | 2 +-
 custom_components/bluetti_bt/switch.py        | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/custom_components/bluetti_bt/binary_sensor.py b/custom_components/bluetti_bt/binary_sensor.py
index 9395da9..a52a269 100644
--- a/custom_components/bluetti_bt/binary_sensor.py
+++ b/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):
diff --git a/custom_components/bluetti_bt/sensor.py b/custom_components/bluetti_bt/sensor.py
index 286f391..9dedc9b 100644
--- a/custom_components/bluetti_bt/sensor.py
+++ b/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))
diff --git a/custom_components/bluetti_bt/switch.py b/custom_components/bluetti_bt/switch.py
index 1f44dc4..1ca2c69 100644
--- a/custom_components/bluetti_bt/switch.py
+++ b/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))