108 lines
3.7 KiB
Python
108 lines
3.7 KiB
Python
import urllib
|
|
import json
|
|
import logging
|
|
|
|
import requests
|
|
from django.conf import settings
|
|
|
|
THEAUDIODB_API_KEY = getattr(settings, "THEAUDIODB_API_KEY")
|
|
ARTIST_SEARCH_URL = f"https://www.theaudiodb.com/api/v1/json/{THEAUDIODB_API_KEY}/search.php?s="
|
|
ARTIST_FETCH_URL = f"https://www.theaudiodb.com/api/v1/json/{THEAUDIODB_API_KEY}/artist.php?i="
|
|
ALBUM_SEARCH_URL = f"https://www.theaudiodb.com/api/v1/json/{THEAUDIODB_API_KEY}/searchalbum.php?s="
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def lookup_artist_from_tadb(name_or_id: str) -> dict:
|
|
artist_info = {}
|
|
response = None
|
|
|
|
# Look for an ID as an integer
|
|
try:
|
|
tadb_id = int(name_or_id)
|
|
except ValueError:
|
|
tadb_id = None
|
|
|
|
if tadb_id:
|
|
response = requests.get(ARTIST_FETCH_URL + str(tadb_id))
|
|
|
|
if response.status_code != 200:
|
|
logger.warn(f"Bad response from TADB: {response.status_code}")
|
|
return artist_info
|
|
|
|
if not response.content:
|
|
logger.warn(f"Bad content from TADB: {response.content}")
|
|
return artist_info
|
|
|
|
if not response:
|
|
name = urllib.parse.quote(name_or_id)
|
|
response = requests.get(ARTIST_SEARCH_URL + name)
|
|
|
|
if response.status_code != 200:
|
|
logger.warn(f"Bad response from TADB: {response.status_code}")
|
|
return artist_info
|
|
|
|
if not response.content:
|
|
logger.warn(f"Bad content from TADB: {response.content}")
|
|
return artist_info
|
|
|
|
if '{"artists": null}' in str(response.content):
|
|
logger.warn(f"Bad content from TADB: {response.content}")
|
|
return artist_info
|
|
|
|
results = json.loads(response.content)
|
|
if results["artists"]:
|
|
artist = results["artists"][0]
|
|
|
|
artist_info["biography"] = artist.get("strBiographyEN")
|
|
artist_info["genre"] = artist.get("strGenre")
|
|
artist_info["mood"] = artist.get("strMood")
|
|
artist_info["thumb_url"] = artist.get("strArtistThumb")
|
|
|
|
return artist_info
|
|
|
|
|
|
def lookup_album_from_tadb(name: str, artist: str) -> dict:
|
|
album_info = {}
|
|
artist = urllib.parse.quote(artist)
|
|
name = urllib.parse.quote(name)
|
|
response = requests.get("".join([ALBUM_SEARCH_URL, artist, "&a=", name]))
|
|
|
|
if response.status_code != 200:
|
|
logger.warn(f"Bad response from TADB: {response.status_code}")
|
|
return {}
|
|
|
|
if not response.content:
|
|
logger.warn(f"Bad content from TADB: {response.content}")
|
|
return {}
|
|
|
|
results = json.loads(response.content)
|
|
if results["album"]:
|
|
album = results["album"][0]
|
|
|
|
album_info["theaudiodb_id"] = album.get("idAlbum")
|
|
album_info["theaudiodb_description"] = album.get("strDescriptionEN")
|
|
album_info["theaudiodb_genre"] = album.get("strGenre")
|
|
album_info["theaudiodb_style"] = album.get("strStyle")
|
|
album_info["theaudiodb_mood"] = album.get("strMood")
|
|
album_info["theaudiodb_speed"] = album.get("strSpeed")
|
|
album_info["theaudiodb_theme"] = album.get("strTheme")
|
|
album_info["allmusic_id"] = album.get("strAllMusicID")
|
|
album_info["wikipedia_slug"] = album.get("strWikipediaID")
|
|
album_info["discogs_id"] = album.get("strDiscogsID")
|
|
album_info["wikidata_id"] = album.get("strWikidataID")
|
|
album_info["rateyourmusic_id"] = album.get("strRateYourMusicID")
|
|
|
|
if album.get("intYearReleased"):
|
|
album_info["theaudiodb_year_released"] = float(
|
|
album.get("intYearReleased")
|
|
)
|
|
if album.get("intScore"):
|
|
album_info["theaudiodb_score"] = float(album.get("intScore"))
|
|
if album.get("intScoreVotes"):
|
|
album_info["theaudiodb_score_votes"] = int(
|
|
album.get("intScoreVotes")
|
|
)
|
|
|
|
return album_info
|