spotify code cleanup

This commit is contained in:
Crussader 2021-10-28 18:23:47 +04:00 committed by GitHub
parent f7a830dc3a
commit 3b999dc02d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 8 deletions

View File

@ -1,15 +1,20 @@
from .track import Track from .track import Track
class Album: class Album:
"""The base class for a Spotify album""" """The base class for a Spotify album"""
def __init__(self, data: dict) -> None: def __init__(self, data: dict) -> None:
self.name = data['name'] self.name = data["name"]
self.artists = ", ".join(artist["name"] for artist in data['artists']) self.artists = ", ".join(artist["name"] for artist in data["artists"])
self.tracks = [Track(track) for track in data['tracks']['items']] self.tracks = [Track(track) for track in data["tracks"]["items"]]
self.total_tracks = data['total_tracks'] self.total_tracks = data["total_tracks"]
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.Album name={self.name} artists={self.artists} id={self.id} total_tracks={self.total_tracks} tracks={self.tracks}>" return (
f"<Pomice.spotify.Album name={self.name} artists={self.artists} id={self.id} "
f"total_tracks={self.total_tracks} tracks={self.tracks}>"
)