diff --git a/api_calls.py b/api_calls.py index 0ecdfa7..f86f03a 100644 --- a/api_calls.py +++ b/api_calls.py @@ -23,14 +23,23 @@ def get_image(mb_id, ftv_api_key, artist_path): response = requests.get(ftv_api_url) ftv_data =response.json() if not ('status' in ftv_data): - art_url = ftv_data['artistthumb'][0]['url'] - print(art_url) - response = requests.get(art_url) - if response.status_code == 200: - with open(os.path.join(artist_path, 'artist.jpg'), 'wb') as f: - f.write(response.content) + if ('artistthumb' in ftv_data): + art_url = ftv_data['artistthumb'][0]['url'] + print(art_url) + response = requests.get(art_url) + if response.status_code == 200: + with open(os.path.join(artist_path, 'artist.jpg'), 'wb') as f: + f.write(response.content) + elif ('hdmusiclogo' in ftv_data): + art_url = ftv_data['hdmusiclogo'][0]['url'] + response = requests.get(art_url) + if response.status_code == 200: + with open(os.path.join(artist_path, 'artist.png'), 'wb') as f: + f.write(response.content) + else: + print("Error downloading: ", response.status_code) else: - print("Error downloading: ", response.status_code) + print("Thumb not found.") else: error_msg = ftv_data['error message'] print(f"Error: {error_msg}") \ No newline at end of file