Clone/zspotify
1
0
Fork 0
mirror of https://github.com/THIS-IS-NOT-A-BACKUP/zspotify.git synced 2024-11-26 18:03:17 +01:00
Code Issues Packages Projects Releases Wiki Activity

Added support for downloading full shows

Browse Source
This commit is contained in:
Footsiefat 2021-10-22 11:21:56 +13:00
parent 0c07408731
commit 355630ba1c
1 changed files with 46 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
zspotify.py
View File

@ -122,7 +122,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 = regex_input_for_urls( track_id_str, album_id_str, playlist_id_str, episode_id_str, show_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:
@ -138,10 +138,14 @@ def client():
print("\n") print("\n")
elif episode_id_str is not None: elif episode_id_str is not None:
download_episode(episode_id_str) download_episode(episode_id_str)
elif show_id_str is not None:
for episode in get_show_episodes(token, show_id_str):
download_episode(episode)
else: else:
search_text = input("Enter search or URL: ") search_text = input("Enter search or URL: ")
track_id_str, album_id_str, playlist_id_str, episode_id_str = regex_input_for_urls( track_id_str, album_id_str, playlist_id_str, episode_id_str, show_id_str = regex_input_for_urls(
search_text) search_text)
if track_id_str is not None: if track_id_str is not None:
@ -157,6 +161,9 @@ def client():
print("\n") print("\n")
elif episode_id_str is not None: elif episode_id_str is not None:
download_episode(episode_id_str) download_episode(episode_id_str)
elif show_id_str is not None:
for episode in get_show_episodes(token, show_id_str):
download_episode(episode)
else: else:
search(search_text) search(search_text)
wait() wait()
@ -191,6 +198,13 @@ def regex_input_for_urls(search_input):
search_input, search_input,
) )
show_uri_search = re.search(
r"^spotify:show:(?P<ShowID>[0-9a-zA-Z]{22})$", search_input)
show_url_search = re.search(
r"^(https?://)?open\.spotify\.com/show/(?P<ShowID>[0-9a-zA-Z]{22})(\?si=.+?)?$",
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
@ -219,7 +233,14 @@ def regex_input_for_urls(search_input):
else: else:
episode_id_str = None episode_id_str = None
return track_id_str, album_id_str, playlist_id_str, episode_id_str if show_uri_search is not None or show_url_search is not None:
show_id_str = (show_uri_search
if show_uri_search is not None else
show_url_search).group("ShowID")
else:
show_id_str = None
return track_id_str, album_id_str, playlist_id_str, episode_id_str, show_id_str
def get_episode_info(episode_id_str): def get_episode_info(episode_id_str):
@ -230,13 +251,29 @@ def get_episode_info(episode_id_str):
if "error" in info: if "error" in info:
return None, None return None, None
else: else:
return info["show"]["name"], info["name"] return sanitize_data(info["show"]["name"]), sanitize_data(info["name"])
def get_show_episodes(access_token, show_id_str):
episodes = []
headers = {'Authorization': f'Bearer {access_token}'}
resp = requests.get(
f'https://api.spotify.com/v1/shows/{show_id_str}/episodes', headers=headers).json()
for episode in resp["items"]:
episodes.append(episode["id"])
return episodes
def download_episode(episode_id_str): def download_episode(episode_id_str):
global ROOT_PODCAST_PATH global ROOT_PODCAST_PATH
podcast_name, episode_name = get_episode_info(episode_id_str) podcast_name, episode_name = get_episode_info(episode_id_str)
extra_paths = podcast_name + "/"
if podcast_name is None: if podcast_name is None:
print("### SKIPPING: (EPISODE NOT FOUND) ###") print("### SKIPPING: (EPISODE NOT FOUND) ###")
else: else:
@ -245,12 +282,13 @@ def download_episode(episode_id_str):
episode_id = EpisodeId.from_base62(episode_id_str) episode_id = EpisodeId.from_base62(episode_id_str)
stream = SESSION.content_feeder().load( stream = SESSION.content_feeder().load(
episode_id, VorbisOnlyAudioQuality(QUALITY), False, None) episode_id, VorbisOnlyAudioQuality(QUALITY), False, None)
print("### DOWNLOADING ENTIRE PODCAST EPISODE - THIS MAY TAKE A WHILE ###") print("### DOWNLOADING '" + podcast_name + " - " +
episode_name + "' - THIS MAY TAKE A WHILE ###")
if not os.path.isdir(ROOT_PODCAST_PATH): if not os.path.isdir(ROOT_PODCAST_PATH + extra_paths):
os.makedirs(ROOT_PODCAST_PATH) os.makedirs(ROOT_PODCAST_PATH + extra_paths)
with open(ROOT_PODCAST_PATH + filename, 'wb') as file: with open(ROOT_PODCAST_PATH + extra_paths + filename, 'wb') as file:
while True: while True:
byte = stream.input_stream.stream().read(1024 * 1024) byte = stream.input_stream.stream().read(1024 * 1024)
if byte == b'': if byte == b'':