fix: swap "async_write_ha_state" with "schedule_update_ha_state" (#2280)

closes #2256

custom_components/alexa_media/light.py

@@ -271,7 +271,7 @@ async def _set_state(self, power_on, brightness=None, kelvin=None, hs_color=None
         self._requested_state_at = datetime.datetime.now(
             datetime.timezone.utc
         )  # must be set last so that previous getters work properly
-        self.async_write_ha_state()
+        self.schedule_update_ha_state()
 
     async def async_turn_on(self, **kwargs):
         """Turn on."""

custom_components/alexa_media/media_player.py

@@ -389,7 +389,7 @@ async def _refresh_if_no_audiopush(already_refreshed=False):
             return
         if event_serial == self.device_serial_number:
             self._available = True
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
         if "last_called_change" in event:
             if (
                 event_serial == self.device_serial_number
@@ -409,8 +409,8 @@ async def _refresh_if_no_audiopush(already_refreshed=False):
                 self._last_called = True
                 self._last_called_timestamp = event["last_called_change"]["timestamp"]
                 self._last_called_summary = event["last_called_change"].get("summary")
-                if self.hass and self.async_write_ha_state:
-                    self.async_write_ha_state()
+                if self.hass and self.schedule_update_ha_state:
+                    self.schedule_update_ha_state()
                 await self._update_notify_targets()
             else:
                 self._last_called = False
@@ -437,8 +437,8 @@ async def _refresh_if_no_audiopush(already_refreshed=False):
                 self._source_list = self._get_source_list()
                 self._connected_bluetooth = self._get_connected_bluetooth()
                 self._bluetooth_list = self._get_bluetooth_list()
-                if self.hass and self.async_write_ha_state:
-                    self.async_write_ha_state()
+                if self.hass and self.schedule_update_ha_state:
+                    self.schedule_update_ha_state()
         elif "player_state" in event:
             player_state = event["player_state"]
             if event_serial == self.device_serial_number:
@@ -470,12 +470,12 @@ async def _refresh_if_no_audiopush(already_refreshed=False):
                         player_state["volumeSetting"],
                     )
                     self._media_vol_level = player_state["volumeSetting"] / 100
-                    if self.hass and self.async_write_ha_state:
-                        self.async_write_ha_state()
+                    if self.hass and self.schedule_update_ha_state:
+                        self.schedule_update_ha_state()
                 elif "dopplerConnectionState" in player_state:
                     self.available = player_state["dopplerConnectionState"] == "ONLINE"
-                    if self.hass and self.async_write_ha_state:
-                        self.async_write_ha_state()
+                    if self.hass and self.schedule_update_ha_state:
+                        self.schedule_update_ha_state()
                 await _refresh_if_no_audiopush(already_refreshed)
         elif "push_activity" in event:
             if self.state in {MediaPlayerState.IDLE, MediaPlayerState.PAUSED, MediaPlayerState.PLAYING}:
@@ -592,7 +592,7 @@ async def refresh(self, device=None, skip_api: bool = False):
                 ]["last_called"].get("summary")
                 await self._update_notify_targets()
             if skip_api and self.hass:
-                self.async_write_ha_state()
+                self.schedule_update_ha_state()
                 return
             if "MUSIC_SKILL" in self._capabilities:
                 if self._parent_clusters and self.hass:
@@ -734,7 +734,7 @@ async def refresh(self, device=None, skip_api: bool = False):
                         )
                     )
         if self.hass:
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
 
     @property
     def source(self):
@@ -989,7 +989,7 @@ async def async_update(self):
                     self.name,
                 )
         self._last_update = util.utcnow()
-        self.async_write_ha_state()
+        self.schedule_update_ha_state()
 
     @property
     def media_content_type(self):
@@ -1074,7 +1074,7 @@ def shuffle(self):
     def shuffle(self, state):
         """Set the Shuffle state."""
         self._shuffle = state
-        self.async_write_ha_state()
+        self.schedule_update_ha_state()
 
     @property
     def repeat_state(self):
@@ -1085,7 +1085,7 @@ def repeat_state(self):
     def repeat_state(self, state):
         """Set the Repeat state."""
         self._repeat = state
-        self.async_write_ha_state()
+        self.schedule_update_ha_state()
 
     @property
     def supported_features(self):

custom_components/alexa_media/sensor.py

@@ -610,7 +610,7 @@ async def async_update(self):
             self._n_dict = None
         self._process_raw_notifications()
         try:
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
         except NoEntitySpecifiedError:
             pass  # we ignore this due to a harmless startup race condition
 

custom_components/alexa_media/switch.py

@@ -214,7 +214,7 @@ def _handle_event(self, event):
         if "queue_state" in event:
             queue_state = event["queue_state"]
             if queue_state["dopplerId"]["deviceSerialNumber"] == self._client.unique_id:
-                self.async_write_ha_state()
+                self.schedule_update_ha_state()
 
     @_catch_login_errors
     async def _set_switch(self, state, **kwargs):
@@ -233,7 +233,7 @@ async def _set_switch(self, state, **kwargs):
                 self.name,
                 getattr(self._client, self._switch_property),
             )
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
         elif self.should_poll:
             # if we need to poll, refresh media_client
             _LOGGER.debug(
@@ -304,7 +304,7 @@ async def async_update(self):
         except AttributeError:
             pass
         try:
-            self.async_write_ha_state()
+            self.schedule_update_ha_state()
         except NoEntitySpecifiedError:
             pass  # we ignore this due to a harmless startup race condition
 
@@ -368,7 +368,7 @@ def _handle_event(self, event):
                 if state != self.is_on:
                     _LOGGER.debug("Detected %s changed to %s", self, state)
                     setattr(self._client, self._switch_property, state)
-                    self.async_write_ha_state()
+                    self.schedule_update_ha_state()
 
 
 class ShuffleSwitch(AlexaMediaSwitch):
@@ -467,12 +467,12 @@ async def _set_state(self, power_on):
         self._requested_state_at = datetime.datetime.now(
             datetime.timezone.utc
         )  # must be set last so that previous getters work properly
-        self.async_write_ha_state()
+        self.schedule_update_ha_state()
 
     async def async_turn_on(self, **kwargs):
         """Turn on."""
         await self._set_state(True)
 
     async def async_turn_off(self, **kwargs):  # pylint:disable=unused-argument
         """Turn off."""
-        await self._set_state(False)
+        await self._set_state(False)