Merge pull request #7 from vveeps/spotify-playlist-info

This commit is contained in:
Clxud 2021-10-06 17:14:40 -04:00 committed by GitHub
commit 134398dcbd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 22 additions and 3 deletions

View File

@ -286,7 +286,9 @@ class Node:
playlist_info={"name": results.name, "selectedTrack": tracks[0]},
tracks=tracks,
ctx=ctx,
spotify=True
spotify=True,
thumbnail=results.images[0].url,
uri=results.url,
)
except SpotifyException:
@ -324,7 +326,9 @@ class Node:
playlist_info={"name": results.name, "selectedTrack": tracks[0]},
tracks=tracks,
ctx=ctx,
spotify=True
spotify=True,
thumbnail=results.images[0].url,
uri=results.url,
)
except SpotifyException:

View File

@ -57,7 +57,9 @@ class Playlist:
playlist_info: dict,
tracks: list,
ctx: Optional[commands.Context] = None,
spotify: bool = False
spotify: bool = False,
thumbnail: Optional[str] = None,
uri: Optional[str] = None,
):
self.playlist_info = playlist_info
self.tracks_raw = tracks
@ -66,6 +68,9 @@ class Playlist:
self.name = playlist_info.get("name")
self.selected_track = playlist_info.get("selectedTrack")
self._thumbnail = thumbnail
self._uri = uri
if self.spotify:
self.tracks = tracks
else:
@ -81,3 +86,13 @@ class Playlist:
def __repr__(self):
return f"<Pomice.playlist name={self.name!r} track_count={len(self.tracks)}>"
@property
def uri(self) -> Optional[str]:
"""Spotify album/playlist URI, or None if not a Spotify object."""
return self._uri
@property
def thumbnail(self) -> Optional[str]:
"""Spotify album/playlist thumbnail, or None if not a Spotify object."""
return self._thumbnail