diff --git a/pomice/spotify/track.py b/pomice/spotify/track.py index 6fe5632..96af859 100644 --- a/pomice/spotify/track.py +++ b/pomice/spotify/track.py @@ -1,12 +1,18 @@ class Track: """The base class for a Spotify Track""" + def __init__(self, data: dict) -> None: self.name = data['name'] self.artists = ", ".join(artist["name"] for artist in data['artists']) self.length = data['duration_ms'] self.id = data['id'] - self.image = data['album']['images'][0]['url'] if data.get('album') else None + + if data.get("album") and data["album"]["images"]: + self.image = data['album']['images'][0]['url'] + else: + self.image = None + self.uri = data['external_urls']['spotify'] def __repr__(self) -> str: - return f"" \ No newline at end of file + return f""