mirror of https://github.com/afkcodes/zspotify.git
65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
from typing import Optional
|
|
|
|
from librespot.audio.decoders import VorbisOnlyAudioQuality
|
|
from librespot.metadata import EpisodeId
|
|
from tqdm import tqdm
|
|
|
|
from const import NAME, ERROR, SHOW, ITEMS, ID, ROOT_PODCAST_PATH, CHUNK_SIZE
|
|
from utils import sanitize_data, create_download_directory, MusicFormat
|
|
from zspotify import ZSpotify
|
|
|
|
EPISODE_INFO_URL = 'https://api.spotify.com/v1/episodes'
|
|
SHOWS_URL = 'https://api.spotify.com/v1/shows'
|
|
|
|
|
|
def get_episode_info(episode_id_str) -> tuple[Optional[str], Optional[str]]:
|
|
info = ZSpotify.invoke_url(f'{EPISODE_INFO_URL}/{episode_id_str}')
|
|
if ERROR in info:
|
|
return None, None
|
|
return sanitize_data(info[SHOW][NAME]), sanitize_data(info[NAME])
|
|
|
|
|
|
def get_show_episodes(show_id_str) -> list:
|
|
episodes = []
|
|
|
|
resp = ZSpotify.invoke_url(f'{SHOWS_URL}/{show_id_str}/episodes')
|
|
|
|
for episode in resp[ITEMS]:
|
|
episodes.append(episode[ID])
|
|
|
|
return episodes
|
|
|
|
|
|
def download_episode(episode_id) -> None:
|
|
podcast_name, episode_name = get_episode_info(episode_id)
|
|
|
|
extra_paths = podcast_name + '/'
|
|
|
|
if podcast_name is None:
|
|
print('### SKIPPING: (EPISODE NOT FOUND) ###')
|
|
else:
|
|
filename = podcast_name + ' - ' + episode_name
|
|
|
|
episode_id = EpisodeId.from_base62(episode_id)
|
|
stream = ZSpotify.get_content_stream(episode_id, ZSpotify.DOWNLOAD_QUALITY)
|
|
|
|
create_download_directory(ZSpotify.get_config(ROOT_PODCAST_PATH) + extra_paths)
|
|
|
|
total_size = stream.input_stream.size
|
|
with open(ZSpotify.get_config(ROOT_PODCAST_PATH) + extra_paths + filename + MusicFormat.WAV.value,
|
|
'wb') as file, tqdm(
|
|
desc=filename,
|
|
total=total_size,
|
|
unit='B',
|
|
unit_scale=True,
|
|
unit_divisor=1024
|
|
) as bar:
|
|
for _ in range(int(total_size / ZSpotify.get_config(CHUNK_SIZE)) + 1):
|
|
bar.update(file.write(
|
|
stream.input_stream.stream().read(ZSpotify.get_config(CHUNK_SIZE))))
|
|
|
|
# convert_audio_format(ROOT_PODCAST_PATH +
|
|
# extra_paths + filename + '.wav')
|
|
|
|
# related functions that do stuff with the spotify API
|