🔧 fix call in unsafe thread (#1645)

custom_components/xiaomi_miot/__init__.py

@@ -1800,7 +1800,7 @@ def get_properties(self, mapping, update_entity=False, throw=False, **kwargs):
 
         if attrs and update_entity:
             self.update_attrs(attrs, update_subs=True)
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
         return attrs
 
     async def async_get_properties(self, mapping, **kwargs):
@@ -1888,7 +1888,7 @@ def set_miot_property(self, siid, piid, value, did=None, **kwargs):
                     pass
                 elif prop := srv.properties.get(piid):
                     self._state_attrs[prop.full_name] = value
-                    self.async_write_ha_state()
+                    self.schedule_update_ha_state()
         return ret
 
     async def async_set_miot_property(self, siid, piid, value, did=None, **kwargs):
@@ -2370,7 +2370,7 @@ async def async_added_to_hass(self):
     def update_from_parent(self):
         self.update()
         if self.platform:
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
 
     def update(self, data=None):
         attrs = self.parent_attributes
@@ -2407,7 +2407,7 @@ def update_attrs(self, attrs: dict, update_parent=True):
                 getattr(self._parent, 'update_attrs')(attrs or {}, update_parent=False)
         if self.hass and self.platform:
             # don't set state before added to hass
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
         return self._state_attrs
 
     def call_parent(self, method, *args, **kwargs):

custom_components/xiaomi_miot/button.py

@@ -136,5 +136,5 @@ def press(self):
             **self._press_kwargs,
         }
         if ret := self._press_action(**kws):
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
         return ret

custom_components/xiaomi_miot/camera.py

@@ -383,7 +383,7 @@ def get_stream_address(self, **kwargs):
                 self._url_expiration = now + 60 * 4.5
             if self._prop_stream_address:
                 self._last_url = self._prop_stream_address.from_dict(odt)
-                self.async_write_ha_state()
+                self.schedule_update_ha_state()
                 self.async_check_stream_address(self._last_url)
                 if not kwargs.get('scheduled') or self.custom_config('keep_streaming'):
                     self._schedule_stream_refresh()

custom_components/xiaomi_miot/cover.py

@@ -414,7 +414,7 @@ async def _time_changed_cover(self, now):
             self._unsub_listener_cover()
             self._unsub_listener_cover = None
             self._set_position = None
-        self.async_write_ha_state()
+        self.schedule_update_ha_state()
         _LOGGER.debug('cover process %s: %s', self.entity_id, {
             'position': self._position,
             'set_position': self._set_position,

custom_components/xiaomi_miot/media_player.py

@@ -818,7 +818,7 @@ def select_source(self, source):
         if source in self._keycodes:
             ret = self.press_key(source)
             self._attr_app_name = source
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
             return ret
         if source in self.source_list:
             return super().select_source(source)

custom_components/xiaomi_miot/number.py

@@ -122,5 +122,5 @@ def set_native_value(self, value):
         ret = self.call_parent('call_action', self._miot_action, [val])
         if ret:
             self._attr_native_value = val
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
         return ret

custom_components/xiaomi_miot/select.py

@@ -86,7 +86,7 @@ def select_option(self, option):
         if act:
             if ret := self.call_action(act):
                 self._attr_current_option = option
-                self.async_write_ha_state()
+                self.schedule_update_ha_state()
                 self._attr_current_option = None
         return ret
 
@@ -153,7 +153,7 @@ def select_option(self, option):
             ret = self.call_parent('call_action', self._miot_action, pms)
         if ret:
             self._attr_current_option = option
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
         return ret
 
 
@@ -168,7 +168,7 @@ def __init__(self, parent, attr, option=None):
     def update(self, data=None):
         super().update(data)
         self._attr_current_option = self._attr_state
-        self.async_write_ha_state()
+        self.schedule_update_ha_state()
 
     def select_option(self, option):
         """Change the selected option."""
@@ -179,7 +179,7 @@ def select_option(self, option):
             }
             if ret := self._select_option(option, **kws):
                 self._attr_current_option = option
-                self.async_write_ha_state()
+                self.schedule_update_ha_state()
             return ret
         raise NotImplementedError()
 

custom_components/xiaomi_miot/switch.py

@@ -219,7 +219,7 @@ def turn_on(self, **kwargs):
         ret = self.call_parent('call_action', self._miot_action, None if val is None else [val])
         if ret:
             self._state = True
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
             time.sleep(0.5)
             self._state = False
         return ret

custom_components/xiaomi_miot/text.py

@@ -108,7 +108,7 @@ def set_value(self, value):
 
         if ret:
             self._attr_native_value = value
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
             time.sleep(0.5)
             self._attr_native_value = ''
         return ret