nambrogi
/
pomice
mirror of https://github.com/cloudwithax/pomice.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix spotify url issue and handle local tracks properly

This commit is contained in:
vveeps 2021-11-06 23:07:11 +02:00
parent ec02cbe146
commit 81bf0b9756
3 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pomice/spotify/album.py
View File

@ -11,7 +11,7 @@ class Album:
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 = f"https://open.spotify.com/album/{self.id}" self.uri = data["external_urls"]["spotify"]
def __repr__(self) -> str: def __repr__(self) -> str:
return ( return (

2
pomice/spotify/playlist.py
View File

@ -12,7 +12,7 @@ class Playlist:
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 = f"https://open.spotify.com/playlist/{self.id}" self.uri = data["external_urls"]["spotify"]
def __repr__(self) -> str: def __repr__(self) -> str:
return ( return (

7
pomice/spotify/track.py
View File

@ -7,12 +7,15 @@ class Track:
self.length = data["duration_ms"] self.length = data["duration_ms"]
self.id = data["id"] self.id = data["id"]
if data.get("album") and data["album"]["images"]: if data.get("album") and data["album"].get("images"):
self.image = data["album"]["images"][0]["url"] self.image = data["album"]["images"][0]["url"]
else: else:
self.image = None self.image = None
self.uri = f"https://open.spotify.com/track/{self.id}" if data["is_local"]:
self.uri = None
else:
self.uri = data["external_urls"]["spotify"]
def __repr__(self) -> str: def __repr__(self) -> str:
return ( return (