Fix T5 actions error in some cases #1328

AlexxIT · Feb 13, 2024 · ba2ebc7 · ba2ebc7
1 parent aa88006
commit ba2ebc7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/custom_components/sonoff/sensor.py b/custom_components/sonoff/sensor.py
@@ -322,7 +322,8 @@ def set_state(self, params: dict):
     async def clear_state(self):
         await asyncio.sleep(0.5)
         self._attr_native_value = ""
-        self._async_write_ha_state()
+        if self.hass:
+            self._async_write_ha_state()
 
 
 class XT5Action(XEntity, SensorEntity):
@@ -346,7 +347,8 @@ def set_state(self, params: dict):
     async def clear_state(self):
         await asyncio.sleep(0.5)
         self._attr_native_value = ""
-        self._async_write_ha_state()
+        if self.hass:
+            self._async_write_ha_state()
 
 
 class XUnknown(XEntity, SensorEntity):