better organization

src/resources/singlelistapi.py

@@ -0,0 +1,111 @@
+"""
+Endpoinds manipulating single lists
+"""
+
+__author__ = '@tormakris'
+__copyright__ = "Copyright 2020, onSpot Team"
+__module_name__ = "singlelistapi"
+__version__text__ = "1"
+
+import musicbrainzngs
+from flask import request, current_app, abort
+
+from flaskaddons.fred import flaskred
+from resources.spotifyuserstoreresource import SpotifyUserStoreResource
+
+
+class SingleListApi(SpotifyUserStoreResource):
+    """
+    See: https://swagger.kmlabz.com/?urls.primaryName=onSpot%20Backend#/backend/getList
+    """
+
+    def get(self, listid: str):
+        try:
+            currcreds = self.encryptor.load(flaskred.get(request.headers.get('Authorization')).decode('UTF-8'))
+        except Exception as e:
+            current_app.logger.warning(e)
+            abort(401, "unauthorized")
+        try:
+            list_type = flaskred.get(listid).decode('UTF-8')
+        except Exception as e:
+            current_app.logger.warning(e)
+            abort(404, "unknown list")
+        musicbrainzngs.auth(currcreds['name'], currcreds['password'])
+        limit = int(request.args.get('limit', 10))
+        offset = int(request.args.get('offset', 0))
+        if list_type == 'release':
+            currdata = musicbrainzngs.get_releases_in_collection(listid, limit, offset)['collection']
+            releaselist = []
+            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']
+                try:
+                    imgurl = musicbrainzngs.get_image_list(release['id'])['images']
+                    if len(imgurl) > 0:
+                        releasedata['cover_url'] = imgurl[0]['image']
+                except Exception as e:
+                    current_app.logger.warning(e)
+                releaselist.append(releasedata)
+                flaskred.set(release['id'], 'release'.encode('UTF-8'))
+            retdata = {"id": currdata['id'], "element_count": currdata['release-count'], "itemlist": releaselist}
+        elif list_type == 'artist':
+            currdata = musicbrainzngs.get_artists_in_collection(listid, limit, offset)['collection']
+            artistlist = []
+            for artist in currdata['artist-list']:
+                artist_data = {"id": artist['id'], "artist": artist['name']}
+                artist_spot = self.spotify.search(q=artist['name'], type="artist", limit=1)['artists']['items']
+                if len(artist_spot) > 0:
+                    artist_data['spotify_id'] = f"spotify:artist:{artist_spot[0]['id']}"
+                    artist_image = artist_spot[0]['images']
+                    if len(artist_image) > 0:
+                        artist_data['cover_url'] = artist_image[0]['url']
+                artistlist.append(artist_data)
+            flaskred.set(artist['id'], 'artist'.encode('UTF-8'))
+            retdata = {"id": currdata['id'], "element_count": currdata['artist-count'], "itemlist": artistlist}
+        elif list_type == 'work':
+            currdata = musicbrainzngs.get_works_in_collection(listid, limit, offset)['collection']
+            worklist = []
+            for work in currdata['work-list']:
+                workdata = {"id": work['id'], "title": work['title']}
+                work_spot = self.spotify.search(q=work['title'], type="track", limit=1)['tracks']['items']
+                if len(work_spot) > 0:
+                    workdata['spotify_id'] = f"spotify:track:{work_spot[0]['id']}"
+                worklist.append(workdata)
+                flaskred.set(work['id'], 'recording'.encode('UTF-8'))
+            retdata = {"id": currdata['id'], "element_count": currdata['work-count'], "itemlist": worklist}
+        elif list_type == 'recording':
+            currdata = musicbrainzngs.get_recordings_in_collection(listid, limit, offset)['collection']
+            recordinglist = []
+            for recording in currdata['recording-list']:
+                currrec = {"id": recording['id'], "title": recording['title']}
+                currrecording = musicbrainzngs.get_recording_by_id(recording['id'], includes=['artists', 'releases'])
+                if 'artist-credit' in currrecording['recording'] and currrecording['recording']['artist-credit']:
+                    currartist = currrecording['recording']['artist-credit'][0]['artist']
+                    currrec['artist'] = currartist['name']
+                if 'release-list' in currrecording['recording'] and currrecording['recording']['release-list']:
+                    currrlease = currrecording['recording']['release-list'][0]
+                    currrec['album'] = currrlease['title']
+                    try:
+                        imgurl = musicbrainzngs.get_image_list(recording['id'])['images']
+                        if len(imgurl) > 0:
+                            currrec['cover_url'] = imgurl[0]['image']
+                    except Exception as e:
+                        current_app.logger.warning(e)
+                recording_spot = self.spotify.search(q=recording['title'], type="track", limit=1)['tracks']['items']
+                if len(recording_spot) > 0:
+                    currrec['spotify_id'] = f"spotify:track:{recording_spot[0]['id']}"
+                recordinglist.append(currrec)
+                flaskred.set(recording['id'], 'recording'.encode('UTF-8'))
+            retdata = {"id": currdata['id'], "element_count": currdata['recording-count'], "itemlist": recordinglist}
+        else:
+            abort(417, "wrong type of collection")
+        musicbrainzngs.auth(None, None)
+        retdata['type'] = list_type
+        retdata['name'] = currdata['name']
+        return retdata, 200