Clone/zspotify
1
0
Fork 0
mirror of https://github.com/THIS-IS-NOT-A-BACKUP/zspotify.git synced 2025-07-01 15:23:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed crash from variable reference before assignment

Browse Source
This commit is contained in:
Footsiefat 2021-10-25 16:44:42 +13:00
parent c4ff812ac3
commit 0b51951b5f
3 changed files with 46 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -151,3 +151,6 @@ ZSpotify\ Podcasts/
# Intellij # Intellij
.idea .idea
#Configuration json file
zs_config.json

20
zs_config.json
View File

@ -1,11 +1,11 @@
{ {
"ROOT_PATH": "../ZSpotify Music/", "ROOT_PATH": "../ZSpotify Music/",
"ROOT_PODCAST_PATH": "../ZSpotify Podcasts/", "ROOT_PODCAST_PATH": "../ZSpotify Podcasts/",
"SKIP_EXISTING_FILES": true, "SKIP_EXISTING_FILES": true,
"DOWNLOAD_FORMAT": "mp3", "DOWNLOAD_FORMAT": "mp3",
"FORCE_PREMIUM": false, "FORCE_PREMIUM": false,
"ANTI_BAN_WAIT_TIME": 1, "ANTI_BAN_WAIT_TIME": 1,
"OVERRIDE_AUTO_WAIT": false, "OVERRIDE_AUTO_WAIT": false,
"CHUNK_SIZE": 50000, "CHUNK_SIZE": 50000,
"SPLIT_ALBUM_DISCS": false "SPLIT_ALBUM_DISCS": false
} }

64
zspotify/app.py
View File

@ -33,12 +33,14 @@ def client() -> None:
elif sys.argv[1] == '-ls' or sys.argv[1] == '--liked-songs': elif sys.argv[1] == '-ls' or sys.argv[1] == '--liked-songs':
for song in get_saved_tracks(): for song in get_saved_tracks():
if not song[TRACK][NAME]: if not song[TRACK][NAME]:
print('### SKIPPING: SONG DOES NOT EXIST ON SPOTIFY ANYMORE ###') print(
'### SKIPPING: SONG DOES NOT EXIST ON SPOTIFY ANYMORE ###')
else: else:
download_track(song[TRACK][ID], 'Liked Songs/') download_track(song[TRACK][ID], 'Liked Songs/')
print('\n') print('\n')
else: else:
track_id, album_id, playlist_id, episode_id, show_id, artist_id = regex_input_for_urls(sys.argv[1]) track_id, album_id, playlist_id, episode_id, show_id, artist_id = regex_input_for_urls(
sys.argv[1])
if track_id is not None: if track_id is not None:
download_track(track_id) download_track(track_id)
@ -51,7 +53,7 @@ def client() -> None:
name, _ = get_playlist_info(playlist_id) name, _ = get_playlist_info(playlist_id)
for song in playlist_songs: for song in playlist_songs:
download_track(song[TRACK][ID], download_track(song[TRACK][ID],
sanitize_data(name) + '/') sanitize_data(name) + '/')
print('\n') print('\n')
elif episode_id is not None: elif episode_id is not None:
download_episode(episode_id) download_episode(episode_id)
@ -64,7 +66,8 @@ def client() -> None:
while len(search_text) == 0: while len(search_text) == 0:
search_text = input('Enter search or URL: ') search_text = input('Enter search or URL: ')
track_id, album_id, playlist_id, episode_id, show_id, artist_id = regex_input_for_urls(search_text) track_id, album_id, playlist_id, episode_id, show_id, artist_id = regex_input_for_urls(
search_text)
if track_id is not None: if track_id is not None:
download_track(track_id) download_track(track_id)
@ -114,7 +117,6 @@ def search(search_term):
raise ValueError('Invalid limit passed. Max is 50.\n') raise ValueError('Invalid limit passed. Max is 50.\n')
params['limit'] = splits[index+1] params['limit'] = splits[index+1]
if split == '-t' or split == '-type': if split == '-t' or split == '-type':
allowed_types = ['track', 'playlist', 'album', 'artist'] allowed_types = ['track', 'playlist', 'album', 'artist']
@ -148,6 +150,7 @@ def search(search_term):
counter = 1 counter = 1
dics = [] dics = []
total_tracks = 0
if TRACK in params['type'].split(','): if TRACK in params['type'].split(','):
tracks = resp[TRACKS][ITEMS] tracks = resp[TRACKS][ITEMS]
if len(tracks) > 0: if len(tracks) > 0:
@ -162,20 +165,20 @@ def search(search_term):
track_data.append([counter, f'{track[NAME]} {explicit}', track_data.append([counter, f'{track[NAME]} {explicit}',
','.join([artist[NAME] for artist in track[ARTISTS]])]) ','.join([artist[NAME] for artist in track[ARTISTS]])])
dics.append({ dics.append({
ID : track[ID], ID: track[ID],
NAME : track[NAME], NAME: track[NAME],
'type' : TRACK, 'type': TRACK,
}) })
counter += 1 counter += 1
total_tracks = counter - 1 total_tracks = counter - 1
print(tabulate(track_data, headers=['S.NO', 'Name', 'Artists'], tablefmt='pretty')) print(tabulate(track_data, headers=[
'S.NO', 'Name', 'Artists'], tablefmt='pretty'))
print('\n') print('\n')
del tracks del tracks
del track_data del track_data
else:
total_tracks = 0
total_albums = 0
if ALBUM in params['type'].split(','): if ALBUM in params['type'].split(','):
albums = resp[ALBUMS][ITEMS] albums = resp[ALBUMS][ITEMS]
if len(albums) > 0: if len(albums) > 0:
@ -185,20 +188,20 @@ def search(search_term):
album_data.append([counter, album[NAME], album_data.append([counter, album[NAME],
','.join([artist[NAME] for artist in album[ARTISTS]])]) ','.join([artist[NAME] for artist in album[ARTISTS]])])
dics.append({ dics.append({
ID : album[ID], ID: album[ID],
NAME : album[NAME], NAME: album[NAME],
'type' : ALBUM, 'type': ALBUM,
}) })
counter += 1 counter += 1
total_albums = counter - total_tracks - 1 total_albums = counter - total_tracks - 1
print(tabulate(album_data, headers=['S.NO', 'Album', 'Artists'], tablefmt='pretty')) print(tabulate(album_data, headers=[
'S.NO', 'Album', 'Artists'], tablefmt='pretty'))
print('\n') print('\n')
del albums del albums
del album_data del album_data
else:
total_albums = 0
total_artists = 0
if ARTIST in params['type'].split(','): if ARTIST in params['type'].split(','):
artists = resp[ARTISTS][ITEMS] artists = resp[ARTISTS][ITEMS]
if len(artists) > 0: if len(artists) > 0:
@ -207,39 +210,39 @@ def search(search_term):
for artist in artists: for artist in artists:
artist_data.append([counter, artist[NAME]]) artist_data.append([counter, artist[NAME]])
dics.append({ dics.append({
ID : artist[ID], ID: artist[ID],
NAME : artist[NAME], NAME: artist[NAME],
'type' : ARTIST, 'type': ARTIST,
}) })
counter += 1 counter += 1
total_artists = counter - total_tracks - total_albums - 1 total_artists = counter - total_tracks - total_albums - 1
print(tabulate(artist_data, headers=['S.NO', 'Name'], tablefmt='pretty')) print(tabulate(artist_data, headers=[
'S.NO', 'Name'], tablefmt='pretty'))
print('\n') print('\n')
del artists del artists
del artist_data del artist_data
else:
total_artists = 0
total_playlists = 0
if PLAYLIST in params['type'].split(','): if PLAYLIST in params['type'].split(','):
playlists = resp[PLAYLISTS][ITEMS] playlists = resp[PLAYLISTS][ITEMS]
if len(playlists) > 0: if len(playlists) > 0:
print('### PLAYLISTS ###') print('### PLAYLISTS ###')
playlist_data = [] playlist_data = []
for playlist in playlists: for playlist in playlists:
playlist_data.append([counter, playlist[NAME], playlist[OWNER][DISPLAY_NAME]]) playlist_data.append(
[counter, playlist[NAME], playlist[OWNER][DISPLAY_NAME]])
dics.append({ dics.append({
ID : playlist[ID], ID: playlist[ID],
NAME : playlist[NAME], NAME: playlist[NAME],
'type' : PLAYLIST, 'type': PLAYLIST,
}) })
counter += 1 counter += 1
total_playlists = counter - total_artists - total_tracks - total_albums - 1 total_playlists = counter - total_artists - total_tracks - total_albums - 1
print(tabulate(playlist_data, headers=['S.NO', 'Name', 'Owner'], tablefmt='pretty')) print(tabulate(playlist_data, headers=[
'S.NO', 'Name', 'Owner'], tablefmt='pretty'))
print('\n') print('\n')
del playlists del playlists
del playlist_data del playlist_data
else:
total_playlists = 0
if total_tracks + total_albums + total_artists + total_playlists == 0: if total_tracks + total_albums + total_artists + total_playlists == 0:
print('NO RESULTS FOUND - EXITING...') print('NO RESULTS FOUND - EXITING...')
@ -261,4 +264,3 @@ def search(search_term):
download_artist_albums(dic[ID]) download_artist_albums(dic[ID])
else: else:
download_playlist(dic) download_playlist(dic)