Handle empty restore state

Fixes #23
jjlawren · Feb 21, 2022 · db6e2b1 · db6e2b1
1 parent 5eecdb8
commit db6e2b1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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