rdelaage
/
zspotify
mirror of https://github.com/afkcodes/zspotify.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplifed the crash catching code a lot

This commit is contained in:
Footsiefat 2021-10-21 14:41:22 +13:00
parent 775d7cf3b6
commit a66848fe8e
1 changed files with 37 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
zspotify.py
View File

@ -468,54 +468,53 @@ def get_saved_tracks(access_token):
def download_track(track_id_str: str, extra_paths=""): def download_track(track_id_str: str, extra_paths=""):
""" Downloads raw song audio from Spotify """ """ Downloads raw song audio from Spotify """
global ROOT_PATH, SKIP_EXISTING_FILES, MUSIC_FORMAT, RAW_AUDIO_AS_IS global ROOT_PATH, SKIP_EXISTING_FILES, MUSIC_FORMAT, RAW_AUDIO_AS_IS
try:
artists, album_name, name, image_url, release_year, disc_number, track_number, scraped_song_id, is_playable = get_song_info(
track_id_str)
track_id = TrackId.from_base62(track_id_str) song_name = artists[0] + " - " + name
artists, album_name, name, image_url, release_year, disc_number, track_number, scraped_song_id, is_playable = get_song_info( filename = ROOT_PATH + extra_paths + song_name + '.' + MUSIC_FORMAT
track_id_str) except Exception as e:
print("### SKIPPING SONG - FAILED TO QUERY METADATA ###")
song_name = artists[0] + " - " + name # print(e)
filename = ROOT_PATH + extra_paths + song_name + '.' + MUSIC_FORMAT
if not is_playable:
print("### SKIPPING:", song_name, "(SONG IS UNAVAILABLE) ###")
else: else:
if os.path.isfile(filename) and SKIP_EXISTING_FILES: try:
print("### SKIPPING:", song_name, "(SONG ALREADY EXISTS) ###") if not is_playable:
else:
if track_id_str != scraped_song_id:
print("### APPLYING PATCH TO LET SONG DOWNLOAD ###")
track_id_str = scraped_song_id
track_id = TrackId.from_base62(track_id_str)
print("### FOUND SONG:", song_name, " ###")
try:
stream = SESSION.content_feeder().load(
track_id, VorbisOnlyAudioQuality(QUALITY), False, None)
except:
print("### SKIPPING:", song_name, print("### SKIPPING:", song_name,
"(GENERAL DOWNLOAD ERROR) ###") "(SONG IS UNAVAILABLE) ###")
else: else:
print("### DOWNLOADING RAW AUDIO ###") if os.path.isfile(filename) and SKIP_EXISTING_FILES:
print("### SKIPPING:", song_name,
"(SONG ALREADY EXISTS) ###")
else:
if track_id_str != scraped_song_id:
track_id_str = scraped_song_id
if not os.path.isdir(ROOT_PATH + extra_paths): track_id = TrackId.from_base62(track_id_str)
os.makedirs(ROOT_PATH + extra_paths) print("### FOUND SONG:", song_name, " ###")
with open(filename, 'wb') as file: stream = SESSION.content_feeder().load(
# Try's to download the entire track at once now to be more efficient. track_id, VorbisOnlyAudioQuality(QUALITY), False, None)
byte = stream.input_stream.stream().read(-1) print("### DOWNLOADING RAW AUDIO ###")
file.write(byte)
if not RAW_AUDIO_AS_IS: if not os.path.isdir(ROOT_PATH + extra_paths):
try: os.makedirs(ROOT_PATH + extra_paths)
with open(filename, 'wb') as file:
# Try's to download the entire track at once now to be more efficient.
byte = stream.input_stream.stream().read(-1)
file.write(byte)
if not RAW_AUDIO_AS_IS:
convert_audio_format(filename) convert_audio_format(filename)
except:
os.remove(filename)
print("### SKIPPING:", song_name,
"(GENERAL CONVERSION ERROR) ###")
else:
set_audio_tags(filename, artists, name, album_name, set_audio_tags(filename, artists, name, album_name,
release_year, disc_number, track_number) release_year, disc_number, track_number)
set_music_thumbnail(filename, image_url) set_music_thumbnail(filename, image_url)
except:
print("### SKIPPING:", song_name,
"(GENERAL DOWNLOAD ERROR) ###")
if os.path.exists(filename):
os.remove(filename)
def download_album(album): def download_album(album):