Clone/zspotify
1
0
Fork 0
mirror of https://github.com/THIS-IS-NOT-A-BACKUP/zspotify.git synced 2025-11-04 05:20:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed bug preventing downloading with artist url

Browse Source
This commit is contained in:
Logykk 2021-10-23 13:00:46 +13:00 committed by GitHub
parent 42a92fdebd
commit d9d18020d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
zspotify.py
View File

@ -143,7 +143,7 @@ def client():
download_track(song['track']['id'], "Liked Songs/") download_track(song['track']['id'], "Liked Songs/")
print("\n") print("\n")
else: else:
track_id_str, album_id_str, playlist_id_str, episode_id_str, show_id_str = regex_input_for_urls( track_id_str, album_id_str, playlist_id_str, episode_id_str, show_id_str, artist_id_str = regex_input_for_urls(
sys.argv[1]) sys.argv[1])
if track_id_str is not None: if track_id_str is not None:
@ -237,7 +237,6 @@ def regex_input_for_urls(search_input):
search_input, search_input,
) )
if track_uri_search is not None or track_url_search is not None: if track_uri_search is not None or track_url_search is not None:
track_id_str = (track_uri_search track_id_str = (track_uri_search
if track_uri_search is not None else if track_uri_search is not None else
@ -585,6 +584,8 @@ def get_album_name(access_token, album_id):
return resp['artists'][0]['name'], sanitize_data(resp['name']) return resp['artists'][0]['name'], sanitize_data(resp['name'])
# Extra functions directly related to spotify artists # Extra functions directly related to spotify artists
def get_artist_albums(access_token, artist_id): def get_artist_albums(access_token, artist_id):
""" Returns artist's albums """ """ Returns artist's albums """
headers = {'Authorization': f'Bearer {access_token}'} headers = {'Authorization': f'Bearer {access_token}'}
@ -594,6 +595,8 @@ def get_artist_albums(access_token, artist_id):
return [resp['items'][i]['id'] for i in range(len(resp['items']))] return [resp['items'][i]['id'] for i in range(len(resp['items']))]
# Extra functions directly related to our saved tracks # Extra functions directly related to our saved tracks
def get_saved_tracks(access_token): def get_saved_tracks(access_token):
""" Returns user's saved tracks """ """ Returns user's saved tracks """
songs = [] songs = []
@ -624,8 +627,10 @@ def download_track(track_id_str: str, extra_paths="", prefix=False, prefix_value
song_name = artists[0] + " - " + name song_name = artists[0] + " - " + name
if prefix: if prefix:
song_name = f'{prefix_value.zfill(2)}-{song_name}' if prefix_value.isdigit() else f'{prefix_value}-{song_name}' song_name = f'{prefix_value.zfill(2)}-{song_name}' if prefix_value.isdigit(
filename = os.path.join(ROOT_PATH, extra_paths, song_name + '.' + MUSIC_FORMAT) ) else f'{prefix_value}-{song_name}'
filename = os.path.join(ROOT_PATH, extra_paths,
song_name + '.' + MUSIC_FORMAT)
except Exception as e: except Exception as e:
print("### SKIPPING SONG - FAILED TO QUERY METADATA ###") print("### SKIPPING SONG - FAILED TO QUERY METADATA ###")
# print(e) # print(e)
@ -686,7 +691,9 @@ def download_album(album):
artist, album_name = get_album_name(token, album) artist, album_name = get_album_name(token, album)
tracks = get_album_tracks(token, album) tracks = get_album_tracks(token, album)
for n, track in tqdm(enumerate(tracks, start=1), unit_scale=True, unit='Song', total=len(tracks)): for n, track in tqdm(enumerate(tracks, start=1), unit_scale=True, unit='Song', total=len(tracks)):
download_track(track['id'], f'{artist}/{album_name}', prefix=True, prefix_value=str(n), disable_progressbar=True) download_track(track['id'], f'{artist}/{album_name}',
prefix=True, prefix_value=str(n), disable_progressbar=True)
def download_artist_albums(artist): def download_artist_albums(artist):
""" Downloads albums of an artist """ """ Downloads albums of an artist """
@ -695,6 +702,7 @@ def download_artist_albums(artist):
for album_id in albums: for album_id in albums:
download_album(album_id) download_album(album_id)
def download_playlist(playlists, playlist_choice): def download_playlist(playlists, playlist_choice):
"""Downloads all the songs from a playlist""" """Downloads all the songs from a playlist"""
token = SESSION.tokens().get("user-read-email") token = SESSION.tokens().get("user-read-email")