diff options
Diffstat (limited to 'code/utils/source_helpers')
| -rw-r--r-- | code/utils/source_helpers/parse.py | 13 | ||||
| -rw-r--r-- | code/utils/source_helpers/spotify/artist.py | 77 |
2 files changed, 89 insertions, 1 deletions
diff --git a/code/utils/source_helpers/parse.py b/code/utils/source_helpers/parse.py index 8489515..b23a895 100644 --- a/code/utils/source_helpers/parse.py +++ b/code/utils/source_helpers/parse.py @@ -7,6 +7,7 @@ from utils.source_helpers.apple import ( ) from utils.source_helpers.spotify import ( album as spotify_album, + artist as spotify_artist, playlist as spotify_playlist, song as spotify_song, ) @@ -29,6 +30,7 @@ async def parse_custom_source( }, "spotify": { "album": spotify_album.load, + "artist": spotify_artist.load, "playlist": spotify_playlist.load, "song": spotify_song.load, }, @@ -43,7 +45,6 @@ async def parse_custom_source( "apple": AppleSource, "spotify": SpotifySource, } - # Catch all songs if "?i=" in query or "/track/" in query: song, embed = await load_funcs[provider]["song"]( @@ -76,5 +77,15 @@ async def parse_custom_source( results = await sources[provider].load_album(self, user, album) else: return None, embed + # Catch Spotify artists + elif "/artist/" in query: + artist, embed = await load_funcs[provider]["artist"]( + headers[provider], query, user + ) + + if artist: + results = await sources[provider].load_artist(self, user, artist) + else: + return None, embed return results, embed diff --git a/code/utils/source_helpers/spotify/artist.py b/code/utils/source_helpers/spotify/artist.py new file mode 100644 index 0000000..995e208 --- /dev/null +++ b/code/utils/source_helpers/spotify/artist.py @@ -0,0 +1,77 @@ +import datetime +import discord +import requests +from typing import Tuple, Optional +from requests.exceptions import JSONDecodeError + +from utils.config import create_embed, LOG + + +async def load( + headers: dict, + query: str, + user: discord.User, +) -> Tuple[Optional[dict], Optional[discord.Embed]]: + """ + Get the artists top tracks from the Spotify API + """ + artist_id = query.split("/artist/")[1] + + try: + # Get the artists songs + response = requests.get( + f"https://api.spotify.com/v1/artists/{artist_id}/top-tracks", + headers=headers, + ) + + if response.status_code == 404: + embed = create_embed( + title="Artist Not Found", + description=( + "Either the provided link is malformed, the artist does" + " not exist, or the artist does not have any songs." + ), + ) + return None, embed + + if response.status_code == 401: + LOG.error( + "Could not authorize with Spotify API. Likely need to" + " restart the bot." + ) + return None, None + + response.raise_for_status() + # Unpack the artists songs + artist = response.json() + name = artist["tracks"][0]["artists"][0]["name"] + num_tracks = len(artist["tracks"]) + + # Get the artist info (for the thumbnail) + response = requests.get( + f"https://api.spotify.com/v1/artists/{artist_id}", + headers=headers, + ) + + response.raise_for_status() + try: + artwork_url = response.json()["images"][0]["url"] + except IndexError: + artwork_url = None + + except IndexError: + LOG.error("Failed unpacking Spotify artist info") + return None, None + except (JSONDecodeError, requests.HTTPError): + LOG.error("Failed making request to Spotify API") + return None, None + + embed = create_embed( + title="Artist Queued", + description=( + f"Top `{num_tracks}` track by **{name}**\n\n" + f"Queued by {user.mention}" + ), + thumbnail=artwork_url, + ) + return artist, embed |