Handling illegal users

Tempoture · Apr 9, 2021 · e4765bf · e4765bf
1 parent f7f1c5e
commit e4765bf
Showing 1 changed file with 16 additions and 16 deletions.
diff --git a/spotify_data/spotify.py b/spotify_data/spotify.py
@@ -178,22 +178,22 @@ def get_all_songs_table_info(access_token,market_id):
             resp =requests.get(req_endpoint, headers=auth_header)
             resp.raise_for_status()
             for track in resp.json()['items']:
-                print(track)
-                songs.add(track['track']['id'])
-                for artist in track['track']['artists']:
-                    artists.add((artist['id'],artist['name']))
-                    if artist['id'] not in artist_songs:
-                        artist_songs[artist['id']] = list()                        
-                    artist_songs[artist['id']].append(track['track']['id'])
-                if playlist['id'] not in songs_playlists:
-                    songs_playlists[playlist['id']] = list()
-                songs_playlists[playlist['id']].append(track['track']['id'])
-                extra_info[track['track']['id']] = {
-                    'name' : track['track']['name'],
-                    'is_explicit' : track['track']['explicit'],
-                    'popularity' : track['track']['popularity'],
-                    'release_date' : track['track']['album']['release_date']
-                }
+                if track: # For Illegal tracks
+                    songs.add(track['track']['id'])
+                    for artist in track['track']['artists']:
+                        artists.add((artist['id'],artist['name']))
+                        if artist['id'] not in artist_songs:
+                            artist_songs[artist['id']] = list()                        
+                        artist_songs[artist['id']].append(track['track']['id'])
+                    if playlist['id'] not in songs_playlists:
+                        songs_playlists[playlist['id']] = list()
+                    songs_playlists[playlist['id']].append(track['track']['id'])
+                    extra_info[track['track']['id']] = {
+                        'name' : track['track']['name'],
+                        'is_explicit' : track['track']['explicit'],
+                        'popularity' : track['track']['popularity'],
+                        'release_date' : track['track']['album']['release_date']
+                    }
         except requests.exceptions.HTTPError as err:
             print("ERR:" + str(err))