nambrogi
/
pomice
mirror of https://github.com/cloudwithax/pomice.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Account for Lavalink v4 changes when loading YT playlists

This commit is contained in:
CorpNewt 2023-09-15 10:35:56 -05:00 committed by GitHub
parent 0904196979
commit d7a7efb051
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pomice/pool.py
View File

@ -813,6 +813,12 @@ class Node:
return None return None
elif load_type in ("PLAYLIST_LOADED", "playlist"): 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 = [ tracks = [
Track( Track(
track_id=track["encoded"], track_id=track["encoded"],
@ -820,10 +826,10 @@ class Node:
ctx=ctx, ctx=ctx,
track_type=TrackType(track["info"]["sourceName"]), track_type=TrackType(track["info"]["sourceName"]),
) )
for track in data[data_type] for track in track_list
] ]
return Playlist( return Playlist(
playlist_info=data["playlistInfo"], playlist_info=playlist_info,
tracks=tracks, tracks=tracks,
playlist_type=PlaylistType(tracks[0].track_type.value), playlist_type=PlaylistType(tracks[0].track_type.value),
thumbnail=tracks[0].thumbnail, thumbnail=tracks[0].thumbnail,