better spotify query
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
2020-11-26 20:48:20 +01:00
parent 410cef1106
commit 9da9e685bf
2 changed files with 21 additions and 12 deletions

View File

@@ -33,8 +33,13 @@ class ItemApi(SpotifyUserStoreResource):
if item_type == 'release':
currrelease = musicbrainzngs.get_release_by_id(itemid, includes=['artists'])['release']
album_spot = self.spotify.search(q=currrelease['title'], type="album", limit=1)['albums']['items']
retdata = {"id": itemid, "album": currrelease['title']}
if 'artist-credit' in currrelease['release'] and currrelease['release']['artist-credit']:
retdata['artist'] = currrelease['artist-credit'][0]['artist']['name']
album_spot = \
self.spotify.search(q=f"{retdata.get('artist', '')} {retdata['title']}", limit=1)[
'albums'][
'items']
try:
imgurl = musicbrainzngs.get_image_list(currrelease['id'])['images']
if len(imgurl) > 0:
@@ -44,8 +49,6 @@ class ItemApi(SpotifyUserStoreResource):
current_app.logger.warning(e)
if len(album_spot) > 0:
retdata['spotify_id'] = f"spotify:album:{album_spot[0]['id']}"
if 'artist-credit' in currrelease and currrelease['artist-credit']:
retdata['artist'] = currrelease['artist-credit'][0]['artist']['name']
elif item_type == 'artist':
currartist = musicbrainzngs.get_artist_by_id(itemid)['artist']
artist_spot = self.spotify.search(q=currartist['name'], type="artist", limit=1)['artists']['items']
@@ -55,7 +58,7 @@ class ItemApi(SpotifyUserStoreResource):
artist_image = artist_spot[0]['images']
if len(artist_image) > 0:
retdata['cover_url'] = artist_image[0]['url']
retdata['cover_url_small'] = artist_image[len(artist_image)-1]['url']
retdata['cover_url_small'] = artist_image[len(artist_image) - 1]['url']
elif item_type == 'work':
currwork = musicbrainzngs.get_work_by_id(itemid)['work']
work_spot = self.spotify.search(q=currwork['title'], type="track", limit=1)['tracks']['items']
@@ -65,10 +68,9 @@ class ItemApi(SpotifyUserStoreResource):
work_image = work_spot[0]['images']
if len(work_image) > 0:
retdata['cover_url'] = work_image[0]['url']
workdata['cover_url_small'] = work_image[len(work_image)-1]['url']
retdata['cover_url_small'] = work_image[len(work_image) - 1]['url']
elif item_type == 'recording':
currrecording = musicbrainzngs.get_recording_by_id(itemid, includes=['artists', 'releases'])['recording']
recording_spot = self.spotify.search(q=currrecording['title'], type="track", limit=1)['tracks']['items']
retdata = {"id": itemid, "title": currrecording['title']}
if 'artist-credit' in currrecording and currrecording['artist-credit']:
currartist = currrecording['artist-credit'][0]['artist']
@@ -83,6 +85,9 @@ class ItemApi(SpotifyUserStoreResource):
retdata['cover_url'] = imgurl[0]['thumbnails']['large']
except Exception as e:
current_app.logger.warning(e)
recording_spot = \
self.spotify.search(q=f"{retdata.get('artist', '')} {retdata.get('album', '')} {retdata['title']}",
limit=1)['tracks']['items']
if len(recording_spot) > 0:
retdata['spotify_id'] = f"spotify:track:{recording_spot[0]['id']}"
else: