onspot/backend
2
0
Fork 0
Code Pull Requests Releases Activity

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

Browse Source
This commit is contained in:
Torma Kristóf
2020-11-26 20:48:20 +01:00
parent 410cef1106
commit 9da9e685bf
2 changed files with 21 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/resources/singlelistapi.py
View File

@ -39,12 +39,14 @@ class SingleListApi(SpotifyUserStoreResource):
for release in currdata['release-list']:
releasedata = {"id": release['id'], "album": release['title']}
currrelease = musicbrainzngs.get_release_by_id(release['id'], includes=['artists'])
album_spot = self.spotify.search(q=release['title'], type="album", limit=1)['albums']['items']
if len(album_spot) > 0:
releasedata['spotify_id'] = f"spotify:album:{album_spot[0]['id']}"
if 'artist-credit' in currrelease['release'] and currrelease['release']['artist-credit']:
currartist = currrelease['release']['artist-credit'][0]['artist']
releasedata['artist'] = currartist['name']
album_spot = \
self.spotify.search(q=f"{releasedata.get('artist', '')} {release['title']}", limit=1)[
'albums']['items']
if len(album_spot) > 0:
releasedata['spotify_id'] = f"spotify:album:{album_spot[0]['id']}"
try:
imgurl = musicbrainzngs.get_image_list(release['id'])['images']
if len(imgurl) > 0:
@ -66,7 +68,7 @@ class SingleListApi(SpotifyUserStoreResource):
artist_image = artist_spot[0]['images']
if len(artist_image) > 0:
artist_data['cover_url'] = artist_image[0]['url']
artist_data['cover_url_small'] = artist_image[len(artist_image)-1]['url']
artist_data['cover_url_small'] = artist_image[len(artist_image) - 1]['url']
artistlist.append(artist_data)
flaskred.set(artist['id'], 'artist'.encode('UTF-8'))
retdata = {"id": currdata['id'], "element_count": currdata['artist-count'], "itemlist": artistlist}
@ -81,7 +83,7 @@ class SingleListApi(SpotifyUserStoreResource):
work_image = work_spot[0]['images']
if len(work_image) > 0:
workdata['cover_url'] = work_image[0]['url']
workdata['cover_url_small'] = work_image[len(work_image)-1]['url']
workdata['cover_url_small'] = work_image[len(work_image) - 1]['url']
worklist.append(workdata)
flaskred.set(work['id'], 'recording'.encode('UTF-8'))
retdata = {"id": currdata['id'], "element_count": currdata['work-count'], "itemlist": worklist}
@ -104,7 +106,9 @@ class SingleListApi(SpotifyUserStoreResource):
currrec['cover_url'] = imgurl[0]['thumbnails']['large']
except Exception as e:
current_app.logger.warning(e)
recording_spot = self.spotify.search(q=recording['title'], type="track", limit=1)['tracks']['items']
recording_spot = \
self.spotify.search(q=f"{currrec.get('artist', '')} {currrec.get('album', '')} {currrec['title']}",
limit=1)['tracks']['items']
if len(recording_spot) > 0:
currrec['spotify_id'] = f"spotify:track:{recording_spot[0]['id']}"
recordinglist.append(currrec)