From db6e2b1a317d4db1fdf108d268fb67e7512ffb48 Mon Sep 17 00:00:00 2001 From: jjlawren Date: Sun, 20 Feb 2022 21:23:04 -0600 Subject: [PATCH] Handle empty restore state Fixes #23 --- custom_components/sonos_cloud/media_player.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/custom_components/sonos_cloud/media_player.py b/custom_components/sonos_cloud/media_player.py index 0f9d1db..13995d3 100644 --- a/custom_components/sonos_cloud/media_player.py +++ b/custom_components/sonos_cloud/media_player.py @@ -53,8 +53,13 @@ def __init__(self, player: dict[str, Any]): async def async_added_to_hass(self): """Complete entity setup.""" await super().async_added_to_hass() + await self.async_restore_states() + + async def async_restore_states(self) -> None: + """Restore last entity state.""" + if (last_state := await self.async_get_last_state()) is None: + return - last_state = await self.async_get_last_state() if volume := last_state.attributes.get("volume_level"): self._attr_volume_level = volume