spotify code cleanup

This commit is contained in:
Crussader 2021-10-28 18:25:38 +04:00 committed by GitHub
parent adadc1f9c4
commit ff1c089a75
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 7 deletions

View File

@ -1,16 +1,21 @@
from .track import Track from .track import Track
from typing import List from typing import List
class Playlist: class Playlist:
"""The base class for a Spotify playlist""" """The base class for a Spotify playlist"""
def __init__(self, data: dict, tracks: List[Track]) -> None: def __init__(self, data: dict, tracks: List[Track]) -> None:
self.name = data['name'] self.name = data["name"]
self.tracks = tracks self.tracks = tracks
self.owner = data['owner']['display_name'] self.owner = data["owner"]["display_name"]
self.total_tracks = data['tracks']['total'] self.total_tracks = data["tracks"]["total"]
self.id = data['id'] self.id = data["id"]
self.image = data['images'][0]['url'] self.image = data["images"][0]["url"]
self.uri = data['external_urls']['spotify'] self.uri = data["external_urls"]["spotify"]
def __repr__(self) -> str: def __repr__(self) -> str:
return f"<Pomice.spotify.Playlist name={self.name} owner={self.owner} id={self.id} total_tracks={self.total_tracks} tracks={self.tracks}>" return (
f"<Pomice.spotify.Playlist name={self.name} owner={self.owner} id={self.id} "
f"total_tracks={self.total_tracks} tracks={self.tracks}>"
)