def _update_episode_data(source_id: str, update_data: dict):
""" Allows to update data for episodes (filtered by source_id)"""
logger.info("[%s] Episodes update data: %s", source_id, update_data)
Episode.update(**update_data).where(
Episode.source_id == source_id,
Episode.status != Episode.STATUS_ARCHIVED).execute()
def download_episode(youtube_link: str, episode_id: int):
""" Allows to download youtube video and recreate specific rss (by requested episode_id) """
episode = Episode.get_by_id(episode_id)
logger.info(
"=== [%s] START downloading process URL: %s FILENAME: %s ===",
episode.source_id,
youtube_link,
episode.file_name,
)
stored_file_size = StorageS3().get_file_size(episode.file_name)
if stored_file_size and stored_file_size == episode.file_size:
logger.info(
"[%s] Episode already downloaded and file correct. Downloading will be ignored.",
episode.source_id,
)
_update_episodes(episode.source_id, stored_file_size)
_update_all_rss(episode.source_id)
return EPISODE_DOWNLOADING_IGNORED
elif episode.status not in (Episode.STATUS_NEW,
Episode.STATUS_DOWNLOADING):
logger.error(
"[%s] Episode is %s but file-size seems not correct. "
"Removing not-correct file %s and reloading it from youtube.",
episode.source_id,
episode.status,
episode.file_name,
)
StorageS3().delete_file(episode.file_name)
logger.info(
"[%s] Mark all episodes with source_id [%s] as downloading.",
episode.source_id,
episode.source_id,
)
query = Episode.update(status=Episode.STATUS_DOWNLOADING).where(
Episode.source_id == episode.source_id,
Episode.status != Episode.STATUS_ARCHIVED,
)
query.execute()
try:
result_filename = youtube_utils.download_audio(youtube_link,
episode.file_name)
except YoutubeException as error:
logger.exception(
"=== [%s] Downloading FAILED: Could not download track: %s. "
"All episodes will be rolled back to NEW state",
episode.source_id,
error,
)
Episode.update(status=Episode.STATUS_NEW).where(
Episode.source_id == episode.source_id).execute()
return EPISODE_DOWNLOADING_ERROR
logger.info("=== [%s] DOWNLOADING was done ===", episode.source_id)
youtube_utils.ffmpeg_preparation(result_filename)
logger.info("=== [%s] POST PROCESSING was done === ", episode.source_id)
# ----- uploading file to cloud -----
remote_url = podcast_utils.upload_episode(result_filename)
if not remote_url:
logger.warning("=== [%s] UPLOADING was broken === ")
_update_episodes(episode.source_id,
file_size=0,
status=Episode.STATUS_ERROR)
return EPISODE_DOWNLOADING_ERROR
_update_episode_data(episode.source_id, {
"file_name": result_filename,
"remote_url": remote_url
})
logger.info("=== [%s] UPLOADING was done === ", episode.source_id)
# -----------------------------------
# ----- update episodes data -------
_update_episodes(episode.source_id,
file_size=StorageS3().get_file_size(result_filename))
_update_all_rss(episode.source_id)
podcast_utils.delete_file(
os.path.join(settings.TMP_AUDIO_PATH, result_filename))
# -----------------------------------
logger.info("=== [%s] DOWNLOADING total finished ===", episode.source_id)
return EPISODE_DOWNLOADING_OK