feat: added additional data

music_kraken/pages/youtube_music/youtube_music.py

@@ -538,6 +538,10 @@ class YoutubeMusic(SuperYouTube):
         for renderer in renderer_list:
             album.add_list_of_other_objects(parse_renderer(renderer))
 
+        for song in album.song_collection:
+            for song_source in song.source_collection:
+                song_source.additional_data["playlist_id"] = browse_id
+
         return album
 
 
@@ -556,10 +560,7 @@ class YoutubeMusic(SuperYouTube):
         uploader = ydl_res.get("uploader", "")
         if uploader.endswith(" - Topic"):
             artist_names = [uploader.rstrip(" - Topic")]
-        """
-        elif "artist" in ydl_res:
-            artist_names = ydl_res.get("artist").split(", ")
-        """ 
+
         artist_list = [
             Artist(
                 name=name,
@@ -577,7 +578,7 @@ class YoutubeMusic(SuperYouTube):
             ))
 
         artist_name = artist_names[0] if len(artist_names) > 0 else None
-        return Song(
+        song = Song(
             title=ydl_res.get("track", clean_song_title(ydl_res.get("title"), artist_name=artist_name)),
             note=ydl_res.get("descriptions"),
             album_list=album_list,
@@ -590,6 +591,41 @@ class YoutubeMusic(SuperYouTube):
             ), source],
         )
 
+        # other song details
+        parsed_url = urlparse(source.url)
+        browse_id = parse_qs(parsed_url.query)['v'][0]
+        request_data = {
+            "captionParams": {},
+            "context": {**self.credentials.context, "adSignalsInfo": {"params": []}},
+            "videoId": browse_id,
+        }
+        if "playlist_id" in source.additional_data:
+            request_data["playlistId"] = source.additional_data["playlist_id"]
+        
+        initial_details = self.yt_music_connection.post(
+            url=get_youtube_url(path="/youtubei/v1/player", query=f"prettyPrint=false"),
+            json=request_data,
+            name=f"fetch_album_{browse_id}.json",
+        )
+
+        if initial_details is None:
+            return song
+
+        dump_to_file(f"fetch_song_{browse_id}.json", initial_details.text, is_json=True, exit_after_dump=False)
+        
+        data = initial_details.json()
+        video_details = data.get("videoDetails", {})
+
+        browse_id = video_details.get("videoId", browse_id)
+        song.title = video_details.get("title", song.title)
+        if video_details.get("isLiveContent", False):
+            for album in song.album_list:
+                album.album_type = AlbumType.LIVE_ALBUM
+        for thumbnail in video_details.get("thumbnails", []):
+            song.artwork.append(**thumbnail)
+
+        return song
+
 
     def fetch_media_url(self, source: Source, ydl_res: dict = None) -> dict:
         def _get_best_format(format_list: List[Dict]) -> dict: