From 67681966eb4d3b52616f8b8795611989a8e3d829 Mon Sep 17 00:00:00 2001 From: Jozef Kruszynski Date: Tue, 5 Nov 2024 20:06:28 +0100 Subject: [PATCH] fix intent after recent refactor --- custom_components/mass/intent.py | 28 +++++++++++++++++++++------- 1 file changed, 21 insertions(+), 7 deletions(-) diff --git a/custom_components/mass/intent.py b/custom_components/mass/intent.py index ba4ce8b8..83180cfc 100644 --- a/custom_components/mass/intent.py +++ b/custom_components/mass/intent.py @@ -94,24 +94,36 @@ async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse return response media_id = json_payload.get(ATTR_MEDIA_ID) media_type = json_payload.get(ATTR_MEDIA_TYPE) - media_item = mass.music.get_item_by_name( - media_id, media_type=MediaType(media_type) - ) + media_items = [] + if isinstance(media_id, list): + media_items = [ + ( + await mass.music.get_item_by_name( + item, media_type=MediaType(media_type) + ) + ).to_dict() + for item in media_id + ] + media_item = media_items + else: + media_item = await mass.music.get_item_by_name( + media_id, media_type=MediaType(media_type) + ) radio_mode = json_payload.get(ATTR_RADIO_MODE, False) else: artist = slots.get(ARTIST_SLOT, {}).get(SLOT_VALUE, "") track = slots.get(TRACK_SLOT, {}).get(SLOT_VALUE, "") album = slots.get(ALBUM_SLOT, {}).get(SLOT_VALUE, "") if track: - media_item = mass.music.get_item_by_name( + media_item = await mass.music.get_item_by_name( track, artist=artist, album=album, media_type=MediaType.TRACK ) elif album: - media_item = mass.music.get_item_by_name( + media_item = await mass.music.get_item_by_name( album, artist=artist, media_type=MediaType.ALBUM ) elif artist: - media_item = mass.music.get_item_by_name( + media_item = await mass.music.get_item_by_name( artist, artist=artist, album=album, media_type=MediaType.ARTIST ) else: @@ -120,7 +132,9 @@ async def async_handle(self, intent_obj: intent.Intent) -> intent.IntentResponse try: await mass.player_queues.play_media( queue_id=mass_player_id, - media=media_item, + media=( + media_item if isinstance(media_item, list) else media_item.to_dict() + ), radio_mode=radio_mode, ) except MusicAssistantError as err: