Kevin-Mok/spotify-lib-vis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix program crash when AudioFeatures is missing

Browse Source 
Fixes #32. AudioFeatures object query is now wrapped in a try/except
block to account for the scenario where the object doesn't exist.
This commit is contained in:
Chris Shyi
2018-06-22 16:28:47 -04:00
parent be815086c0
commit 0714586e07
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
spotifyvis/utils.py
View File

@@ -127,7 +127,7 @@ def save_audio_features(headers, track_id, track):
response = requests.get("https://api.spotify.com/v1/audio-features/{}".format(track_id), headers = headers).json() response = requests.get("https://api.spotify.com/v1/audio-features/{}".format(track_id), headers = headers).json()
if 'error' in response: if 'error' in response:
return {} return
# Data that we don't need # Data that we don't need
useless_keys = [ useless_keys = [

3
spotifyvis/views.py
View File

@@ -204,8 +204,11 @@ def get_audio_feature_data(request, audio_feature, client_secret):
'data_points': [], 'data_points': [],
} }
for track in user_tracks: for track in user_tracks:
try:
audio_feature_obj = AudioFeatures.objects.get(track=track) audio_feature_obj = AudioFeatures.objects.get(track=track)
response_payload['data_points'].append(getattr(audio_feature_obj, audio_feature)) response_payload['data_points'].append(getattr(audio_feature_obj, audio_feature))
except AudioFeatures.DoesNotExist:
continue
return JsonResponse(response_payload) return JsonResponse(response_payload)
# }}} get_audio_feature_data # # }}} get_audio_feature_data #