get_album: don't overwrite parsed artist (closes #446)

sigma67 · Oct 3, 2023 · 9a87990 · 9a87990
1 parent d4a088d
commit 9a87990
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/tests/test.py b/tests/test.py
@@ -212,6 +212,9 @@ def test_get_album(self):
         self.assertEqual(len(results["other_versions"]), 2)
         results = self.yt.get_album("MPREb_BQZvl3BFGay")
         self.assertEqual(len(results["tracks"]), 7)
+        self.assertEqual(len(results["tracks"][0]["artists"]), 1)
+        results = self.yt.get_album("MPREb_rqH94Zr3NN0")
+        self.assertEqual(len(results["tracks"][0]["artists"]), 2)
 
     def test_get_song(self):
         song = self.yt_oauth.get_song(config["uploads"]["private_upload_id"])  # private upload

diff --git a/ytmusicapi/mixins/browsing.py b/ytmusicapi/mixins/browsing.py
@@ -419,7 +419,7 @@ def get_album(self, browseId: str) -> Dict:
         album['duration_seconds'] = sum_total_duration(album)
         for i, track in enumerate(album['tracks']):
             album['tracks'][i]['album'] = album['title']
-            album['tracks'][i]['artists'] = album['artists']
+            album['tracks'][i]['artists'] = album['tracks'][i]['artists'] or album['artists']
 
         return album