parse_playlist_items: only add track_position for albums

sigma67 · Jan 11, 2024 · 9d65913 · 9d65913
1 parent 3aa5227
commit 9d65913
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/ytmusicapi/parsers/playlists.py b/ytmusicapi/parsers/playlists.py
@@ -79,11 +79,6 @@ def parse_playlist_items(results, menu_entries: Optional[List[List]] = None, is_
             True,
         )
 
-        track_position = None
-        if is_album:
-            track_pos_found = nav(data, ["index", "runs", 0, "text"], True)
-            track_position = track_pos_found if track_pos_found is None else int(track_pos_found)
-
         song = {
             "videoId": videoId,
             "title": title,
@@ -96,9 +91,12 @@ def parse_playlist_items(results, menu_entries: Optional[List[List]] = None, is_
             "isExplicit": isExplicit,
             "videoType": videoType,
             "views": views,
-            "track_position": track_position,
         }
 
+        if is_album:
+            track_pos_found = nav(data, ["index", "runs", 0, "text"], True)
+            song["track_position"] = track_pos_found if track_pos_found is None else int(track_pos_found)
+
         if duration:
             song["duration"] = duration
             song["duration_seconds"] = parse_duration(duration)