diff --git a/pomice/pool.py b/pomice/pool.py index 55a2706..dd0aee5 100644 --- a/pomice/pool.py +++ b/pomice/pool.py @@ -813,6 +813,12 @@ class Node: return None elif load_type in ("PLAYLIST_LOADED", "playlist"): + if self._version.major >= 4: + track_list = data[data_type]["tracks"] + playlist_info = data[data_type]["info"] + else: + track_list = data[data_type] + playlist_info = data["playlistInfo"] tracks = [ Track( track_id=track["encoded"], @@ -820,10 +826,10 @@ class Node: ctx=ctx, track_type=TrackType(track["info"]["sourceName"]), ) - for track in data[data_type] + for track in track_list ] return Playlist( - playlist_info=data["playlistInfo"], + playlist_info=playlist_info, tracks=tracks, playlist_type=PlaylistType(tracks[0].track_type.value), thumbnail=tracks[0].thumbnail,