diff --git a/app/api/spotify_handler.py b/app/api/spotify_handler.py index 28da68f..f39538e 100644 --- a/app/api/spotify_handler.py +++ b/app/api/spotify_handler.py @@ -24,14 +24,18 @@ def get_user_playlist_data(self, auth_header, user_id): playlist_data = [] for playlist in playlists: - playlist_data.append({ - 'playlist_name': playlist['name'], - 'playlist_url': playlist['external_urls']['spotify'], - 'playlist_img_url': playlist['images'][0]['url'], - 'playlist_tracks_url': playlist['tracks']['href'], - 'playlist_id': playlist['id'], - 'playlist_tracks': self._get_playlist_tracks(auth_header, playlist['id']) - }) + # error fix for issue #2 + try: + playlist_data.append({ + 'playlist_name': playlist['name'], + 'playlist_url': playlist['external_urls']['spotify'], + 'playlist_img_url': playlist['images'][0]['url'], + 'playlist_tracks_url': playlist['tracks']['href'], + 'playlist_id': playlist['id'], + 'playlist_tracks': self._get_playlist_tracks(auth_header, playlist['id']) + }) + except IndexError: + pass return playlist_data def _get_playlist_tracks(self, auth_header, playlist_id):