danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[BACKEND] Add URL thumbnail download timeout

Browse source
This commit is contained in:
Jesse Bannon 2022-11-26 23:37:14 -08:00
parent a77885f4de
commit f74c477423
2 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/ytdl_sub/downloaders/downloader.py
View file

@ -624,7 +624,7 @@ class Downloader(DownloadArchiver, Generic[DownloaderOptionsT], ABC):
continue continue
if (thumbnail_url := parent.get_thumbnail_url(thumbnail_id=thumbnail_id)) is None: if (thumbnail_url := parent.get_thumbnail_url(thumbnail_id=thumbnail_id)) is None:
download_logger.warning("TODO: Failed to download channel's avatar image") download_logger.warning("Failed to find thumbnail id '%s'", thumbnail_id)
continue continue
if self._download_thumbnail( if self._download_thumbnail(
@ -634,7 +634,7 @@ class Downloader(DownloadArchiver, Generic[DownloaderOptionsT], ABC):
self.save_file(file_name=thumbnail_name) self.save_file(file_name=thumbnail_name)
thumbnails_downloaded.add(thumbnail_name) thumbnails_downloaded.add(thumbnail_name)
else: else:
download_logger.warning("TODO: Failed to download channel's avatar image") download_logger.warning("Failed to download thumbnail id '%s'", thumbnail_id)
return thumbnails_downloaded return thumbnails_downloaded

3
src/ytdl_sub/utils/thumbnail.py
View file

@ -60,7 +60,8 @@ def convert_url_thumbnail(thumbnail_url: str, output_thumbnail_path: str) -> Opt
------- -------
True to indicate it converted the thumbnail from url. None if the retry failed. True to indicate it converted the thumbnail from url. None if the retry failed.
""" """
with urlopen(thumbnail_url) as file: # timeout after 8 seconds
with urlopen(thumbnail_url, timeout=8.0) as file:
with tempfile.NamedTemporaryFile() as thumbnail: with tempfile.NamedTemporaryFile() as thumbnail:
thumbnail.write(file.read()) thumbnail.write(file.read())