From 6eb81c9dd1fc0bdaf40d1ac8e38306fef29d4a40 Mon Sep 17 00:00:00 2001
From: Rafael Moraes <50295204+glomatico@users.noreply.github.com>
Date: Tue, 24 Sep 2024 21:43:44 -0300
Subject: [PATCH] add InquirerPy to `get_music_video_id_from_song_id`

---
 votify/downloader_music_video.py | 44 +++++++++++++++++++++++++++++---
 1 file changed, 41 insertions(+), 3 deletions(-)

diff --git a/votify/downloader_music_video.py b/votify/downloader_music_video.py
index 3af94ab..cebbabc 100644
--- a/votify/downloader_music_video.py
+++ b/votify/downloader_music_video.py
@@ -2,6 +2,9 @@ from __future__ import annotations
 
 import logging
 
+from InquirerPy import inquirer
+from InquirerPy.base.control import Choice
+
 from .downloader_video import DownloaderVideo
 
 logger = logging.getLogger("votify")
@@ -82,7 +85,40 @@ class DownloaderMusicVideo(DownloaderVideo):
         ]
         if not related_music_videos:
             return
-        return related_music_videos[0]["trackOfVideo"]["data"]["uri"].split(":")[-1]
+        choices = [
+            Choice(
+                name="None",
+                value=None,
+            )
+        ]
+        choices.extend(
+            [
+                Choice(
+                    name=" | ".join(
+                        [
+                            self.downloader.get_artist_string(
+                                [
+                                    {"name": artist["profile"]["name"]}
+                                    for artist in related_music_video["trackOfVideo"][
+                                        "data"
+                                    ]["artists"]["items"]
+                                ]
+                            ),
+                            related_music_video["trackOfVideo"]["data"]["name"],
+                        ]
+                    ),
+                    value=related_music_video["trackOfVideo"]["data"]["uri"].split(":")[
+                        -1
+                    ],
+                )
+                for related_music_video in related_music_videos
+            ]
+        )
+        selected_music_video_id = inquirer.select(
+            message="Select which music video to download:",
+            choices=choices,
+        ).execute()
+        return selected_music_video_id
 
     def download(
         self,
@@ -94,7 +130,7 @@ class DownloaderMusicVideo(DownloaderVideo):
         finally:
             if self.downloader.temp_path.exists():
                 logger.debug(f'Cleaning up "{self.downloader.temp_path}"')
-                self.downloader.cleanup_temp_path_ab()
+                self.downloader.cleanup_temp_path()
 
     def _download(
         self,
@@ -124,7 +160,9 @@ class DownloaderMusicVideo(DownloaderVideo):
                 album_metadata["artists"][0]["id"],
             )
             if not music_video_id:
-                logger.warning("No alternative music video ID found, skipping")
+                logger.warning(
+                    "No related music videos found or no music video selected, skipping"
+                )
                 return
             music_video_metadata = self.downloader.spotify_api.get_track(music_video_id)
             logger.warning(