dont error on imageless spotify tracks

This commit is contained in:
vveeps 2021-10-26 17:57:42 +03:00
parent 1b54aae3b3
commit 762160e84b
1 changed files with 8 additions and 2 deletions

View File

@ -1,11 +1,17 @@
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: