110 lines
5.1 KiB
Python
110 lines
5.1 KiB
Python
"""
|
|
Base resource with user handling, Spotify and MusicBrains integration
|
|
"""
|
|
|
|
__author__ = '@tormakris'
|
|
__copyright__ = "Copyright 2020, onSpot Team"
|
|
__module_name__ = "apiinteractionresource"
|
|
__version__text__ = "1"
|
|
|
|
from flask import current_app
|
|
from spotipy.oauth2 import SpotifyClientCredentials
|
|
import spotipy
|
|
import musicbrainzngs
|
|
|
|
from resources.userstoreresource import UserStoreResource
|
|
from utils.config import SPOTIFY_CLIENT_ID, SPOTIFY_CLIENT_SECRET
|
|
|
|
|
|
class APIInteractionResource(UserStoreResource):
|
|
spotify = spotipy.Spotify(auth_manager=SpotifyClientCredentials(client_id=SPOTIFY_CLIENT_ID,
|
|
client_secret=SPOTIFY_CLIENT_SECRET))
|
|
|
|
def queryspotifyinfo(self, mb_type: str, title: str = "", artist: str = "", album: str = "") -> tuple:
|
|
querystring = f"{artist} {album} {title}"
|
|
spotify_uri = "spotify:"
|
|
if mb_type == "artist":
|
|
spot_data = self.spotify.search(q=querystring, type="artist", limit=1)['artists']['items']
|
|
if not spot_data:
|
|
return None, None, None
|
|
spotify_uri += "artist:"
|
|
imagedata = spot_data[0]['images']
|
|
elif mb_type == "release":
|
|
spot_data = self.spotify.search(q=querystring, type="album", limit=1)['albums']['items']
|
|
if not spot_data:
|
|
return None, None, None
|
|
spotify_uri += "album:"
|
|
imagedata = spot_data[0]['images']
|
|
elif mb_type == "work" or mb_type == "recording":
|
|
spot_data = self.spotify.search(q=querystring, type="track", limit=1)['tracks']['items']
|
|
if not spot_data:
|
|
return None, None, None
|
|
spotify_uri += "track:"
|
|
imagedata = spot_data[0]['album']['images']
|
|
else:
|
|
return None, None, None
|
|
|
|
spotify_uri += spot_data[0]['id']
|
|
if imagedata:
|
|
return spotify_uri, imagedata[0]['url'], imagedata[len(imagedata) - 1]['url']
|
|
else:
|
|
return spotify_uri, None, None
|
|
|
|
def getcoverimage(self, releaseid: str, itemtype: str, artist: str, title: str = "",
|
|
album: str = "") -> tuple:
|
|
if itemtype != "recording" and itemtype != "release":
|
|
return None, None
|
|
try:
|
|
musicbimgurl = musicbrainzngs.get_image_list(releaseid)['images']
|
|
imgurl = musicbimgurl[0]['thumbnails']['large']
|
|
smallimgurl = musicbimgurl[0]['thumbnails']['small']
|
|
except Exception as e:
|
|
_, imgurl, smallimgurl = self.queryspotifyinfo(itemtype, title=title, album=album, artist=artist)
|
|
current_app.logger.info(e)
|
|
return imgurl, smallimgurl
|
|
|
|
def getreleaseinfo(self, itemid: str) -> dict:
|
|
currrelease = musicbrainzngs.get_release_by_id(itemid, includes=['artists'])['release']
|
|
retdata = {"id": itemid, "album": currrelease['title'], "type": "release"}
|
|
if 'artist-credit' in currrelease and currrelease['artist-credit']:
|
|
retdata['artist'] = currrelease['artist-credit'][0]['artist']['name']
|
|
retdata['cover_url'], retdata['cover_url_small'] = self.getcoverimage(
|
|
itemid, 'release', album=retdata.get('album'),
|
|
artist=retdata.get('artist'))
|
|
retdata['spotify_id'], _, _ = \
|
|
self.queryspotifyinfo('release', album=currrelease['title'])
|
|
return retdata
|
|
|
|
def getartistinfo(self, itemid: str) -> dict:
|
|
currartist = musicbrainzngs.get_artist_by_id(itemid)['artist']
|
|
retdata = {"id": itemid, "artist": currartist['name'], "type": "artist"}
|
|
retdata['spotify_id'], retdata['cover_url'], retdata['cover_url_small'] = \
|
|
self.queryspotifyinfo('artist', artist=retdata['artist'])
|
|
return retdata
|
|
|
|
def getworkinfo(self, itemid: str) -> dict:
|
|
currwork = musicbrainzngs.get_work_by_id(itemid)['work']
|
|
retdata = {"id": itemid, "title": currwork['title'], "type": "work"}
|
|
retdata['spotify_id'], retdata['cover_url'], retdata['cover_url_small'] = \
|
|
self.queryspotifyinfo('work', title=retdata['title'])
|
|
return retdata
|
|
|
|
def getrecordinginfo(self, itemid: str) -> dict:
|
|
currrecording = musicbrainzngs.get_recording_by_id(itemid, includes=['artists', 'releases'])['recording']
|
|
retdata = {"id": itemid, "title": currrecording['title'], "type": "recording"}
|
|
if 'artist-credit' in currrecording and currrecording['artist-credit']:
|
|
currartist = currrecording['artist-credit'][0]['artist']
|
|
retdata['artist'] = currartist['name']
|
|
if 'release-list' in currrecording and currrecording['release-list']:
|
|
currrlease = currrecording['release-list'][0]
|
|
retdata['album'] = currrlease['title']
|
|
imageid = currrlease['id']
|
|
else:
|
|
imageid = itemid
|
|
retdata['cover_url'], retdata['cover_url_small'] = self.getcoverimage(
|
|
imageid, 'release', title=retdata['title'],
|
|
artist=retdata.get('artist'))
|
|
retdata['spotify_id'], _, _ = \
|
|
self.queryspotifyinfo('recording', title=retdata['title'])
|
|
return retdata
|