Adapt update_ha_state api

Author: chenwenhao

custom_components/sensecraft/image.py

@@ -76,7 +76,8 @@ def handle_event(self, event):
             self._image_path = None
 
         self._attr_image_last_updated = dt_util.utcnow() 
-        self.hass.async_create_task(self.async_write_ha_state)
+        if self.hass:
+            self.hass.loop.call_soon_threadsafe(self.async_schedule_update_ha_state)
 
     @property
     def device_info(self) -> DeviceInfo:

custom_components/sensecraft/number.py

@@ -143,7 +143,8 @@ async def async_set_native_value(self, value: float) -> None:
         local: GroveVisionAiV2 = data[GROVE_VISION_AI_V2]
         if local is not None:
             local.device.tscore = value
-            self.hass.async_create_task(self.async_write_ha_state)
+            if self.hass:
+                self.hass.loop.call_soon_threadsafe(self.async_schedule_update_ha_state)
 
 
 class IOU(ConfigNumber):
@@ -169,7 +170,8 @@ async def async_set_native_value(self, value: float) -> None:
         local: GroveVisionAiV2 = data[GROVE_VISION_AI_V2]
         if local is not None:
             local.device.tiou = value
-            self.hass.async_create_task(self.async_write_ha_state)
+            if self.hass:
+                self.hass.loop.call_soon_threadsafe(self.async_schedule_update_ha_state)
 
 
 class ReCameraMotor(NumberEntity):
@@ -218,11 +220,13 @@ async def async_set_native_value(self, value: float) -> None:
             # 发送控制命令
             self._recamera.send_control(command)
             self._attr_native_value = value
-            self.hass.async_create_task(self.async_write_ha_state)
+            if self.hass:
+                self.hass.loop.call_soon_threadsafe(self.async_schedule_update_ha_state)
         except Exception as e:
             _LOGGER.error("Error setting motor position: %s", e)
 
     def update_state(self, angle: float) -> None:
         """更新电机状态."""
         self._attr_native_value = angle
-        self.hass.async_create_task(self.async_write_ha_state)
+        if self.hass:
+            self.hass.loop.call_soon_threadsafe(self.async_schedule_update_ha_state)

custom_components/sensecraft/sensor.py

@@ -189,7 +189,8 @@ async def async_will_remove_from_hass(self) -> None:
 
     def handle_event(self, event):
         self._state = event.data.get('value')
-        self.hass.async_create_task(self.async_write_ha_state())
+        if self.hass:
+            self.hass.loop.call_soon_threadsafe(self.async_schedule_update_ha_state)
 
     @property
     def device_info(self) -> DeviceInfo:
@@ -242,7 +243,8 @@ async def async_will_remove_from_hass(self) -> None:
 
     def handle_event(self, event):
         self._state = event.data.get('value')
-        self.hass.async_create_task(self.async_write_ha_state())
+        if self.hass:
+            self.hass.loop.call_soon_threadsafe(self.async_schedule_update_ha_state)
 
     @property
     def device_info(self) -> DeviceInfo:
@@ -290,7 +292,8 @@ async def async_will_remove_from_hass(self) -> None:
 
     def handle_event(self, event):
         self._state = event.data.get('value')
-        self.hass.async_create_task(self.async_write_ha_state())
+        if self.hass:
+            self.hass.loop.call_soon_threadsafe(self.async_schedule_update_ha_state)
 
     @property
     def device_info(self) -> DeviceInfo:
@@ -335,7 +338,8 @@ async def async_will_remove_from_hass(self) -> None:
 
     def handle_event(self, event):
         self._state = event.data.get('value')
-        self.hass.async_create_task(self.async_write_ha_state())
+        if self.hass:
+            self.hass.loop.call_soon_threadsafe(self.async_schedule_update_ha_state)
 
     @property
     def device_info(self) -> DeviceInfo:
@@ -379,7 +383,8 @@ async def async_will_remove_from_hass(self) -> None:
 
     def handle_event(self, event):
         self._state = event.data.get('text')
-        self.hass.async_create_task(self.async_write_ha_state())
+        if self.hass:
+            self.hass.loop.call_soon_threadsafe(self.async_schedule_update_ha_state)
 
     @property
     def device_info(self) -> DeviceInfo: