Fix race condition where provider is not yet available at startup

music-assistant · Nov 21, 2024 · 1900922 · 1900922
1 parent da17da1
commit 1900922
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/custom_components/mass/entity.py b/custom_components/mass/entity.py
@@ -26,7 +26,7 @@ def __init__(self, mass: MusicAssistantClient, player_id: str) -> None:
         self.mass = mass
         self.player_id = player_id
         player = mass.players.get(player_id)
-        provider = self.mass.get_provider(player.provider)
+        provider = self.mass.get_provider(player.provider, True)
         if TYPE_CHECKING:
             assert provider is not None
 

diff --git a/custom_components/mass/media_player.py b/custom_components/mass/media_player.py
@@ -64,6 +64,8 @@
     from music_assistant_models.player import Player
     from music_assistant_models.player_queue import PlayerQueue
 
+    from . import MusicAssistantConfigEntry
+
 SUPPORTED_FEATURES = (
     MediaPlayerEntityFeature.PAUSE
     | MediaPlayerEntityFeature.VOLUME_SET
@@ -126,7 +128,7 @@ async def wrapper(
 
 async def async_setup_entry(
     hass: HomeAssistant,
-    entry: MusicAssistantBaseEntity,
+    entry: MusicAssistantConfigEntry,
     async_add_entities: AddEntitiesCallback,
 ) -> None:
     """Set up Music Assistant MediaPlayer(s) from Config Entry."""