get_album (via parse_playlist_items): add track_position key + tests …

…for the badly indexed album

tests/mixins/test_browsing.py

@@ -55,19 +55,23 @@ def test_get_album_browse_id_issue_470(self, yt):
         escaped_browse_id = yt.get_album_browse_id("OLAK5uy_nbMYyrfeg5ZgknoOsOGBL268hGxtcbnDM")
         assert escaped_browse_id == "MPREb_scJdtUCpPE2"
 
-    def test_get_album(self, yt, yt_auth, sample_album):
+    def test_get_album(self, yt, yt_auth, sample_album, badly_indexed_album):
         results = yt_auth.get_album(sample_album)
         assert len(results) >= 9
         assert results["tracks"][0]["isExplicit"]
         assert all(item["views"] is not None for item in results["tracks"])
         assert all(item["album"] is not None for item in results["tracks"])
+        assert results["tracks"][0]["track_position"] == 1
         assert "feedbackTokens" in results["tracks"][0]
         assert len(results["other_versions"]) >= 1  # appears to be regional
         results = yt.get_album("MPREb_BQZvl3BFGay")
         assert len(results["tracks"]) == 7
         assert len(results["tracks"][0]["artists"]) == 1
         results = yt.get_album("MPREb_rqH94Zr3NN0")
         assert len(results["tracks"][0]["artists"]) == 2
+        results = yt.get_album(badly_indexed_album)  # album with non-standard indexing
+        assert results["tracks"][0]["track_position"] == 3
+        assert results["tracks"][13]["track_position"] == 18
 
     def test_get_song(self, config, yt, yt_oauth, sample_video):
         song = yt_oauth.get_song(config["uploads"]["private_upload_id"])  # private upload

ytmusicapi/parsers/playlists.py

@@ -79,6 +79,12 @@ def parse_playlist_items(results, menu_entries: Optional[List[List]] = None):
             True,
         )
 
+        track_position = None
+        if "index" in data:
+            # similar item nesting to duration above -> both could be refactored into
+            # a function similar to nav (?)
+            track_position = int(data["index"]["runs"][0]["text"])
+
         song = {
             "videoId": videoId,
             "title": title,
@@ -91,7 +97,9 @@ def parse_playlist_items(results, menu_entries: Optional[List[List]] = None):
             "isExplicit": isExplicit,
             "videoType": videoType,
             "views": views,
+            "track_position": track_position,
         }
+
         if duration:
             song["duration"] = duration
             song["duration_seconds"] = parse_duration(duration)