src/resources/apiinteractionresource.py

@@ -25,23 +25,30 @@ class APIInteractionResource(UserStoreResource):
         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="artist", limit=1).get('albums')
-            if not spot_data:
+            if spot_data:
+                imagedata = spot_data[0]['images']
+            else:
+                return None, None, None
+        elif mb_type == "release":
+            spot_data = self.spotify.search(q=querystring, type="artist", limit=1).get(
+                'albums')
+            if spot_data:
+                spot_data = spot_data['items']
+            else:
                 return None, None, None
-            spot_data = spot_data['items']
             spotify_uri += "album:"
-            imagedata = spot_data[0]['images']
+            if spot_data:
+                imagedata = spot_data[0]['images']
+            else:
+                return None, None, None
         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']
+            if spot_data:
+                imagedata = spot_data[0]['album']['images']
+            else:
+                return None, None, None
         else:
             return None, None, None