[view.core] fix: not handling exceptions when downloading a file

This commit is contained in:
Vinicius Moreira
2023-12-14 14:17:57 -03:00
parent e44db92a1d
commit 0bc66a9d16

View File

@@ -73,11 +73,16 @@ class SelfFileDownloader(FileDownloader):
known_size = content_length and content_length > 0 known_size = content_length and content_length > 0
total_size_str = get_human_size_str(content_length) if known_size > 0 else "?" total_size_str = get_human_size_str(content_length) if known_size > 0 else "?"
for data in res.iter_content(chunk_size=1024): try:
byte_stream.write(data) for data in res.iter_content(chunk_size=1024):
total_downloaded += len(data) byte_stream.write(data)
perc = f"({(total_downloaded / content_length) * 100:.2f}%) " if known_size > 0 else "" total_downloaded += len(data)
watcher.change_substatus(f"{perc}{base_msg} ({get_human_size_str(total_downloaded)} / {total_size_str})") perc = f"({(total_downloaded / content_length) * 100:.2f}%) " if known_size > 0 else ""
watcher.change_substatus(f"{perc}{base_msg} ({get_human_size_str(total_downloaded)} / {total_size_str})")
except Exception:
self._logger.error(f"Unexpected exception while downloading file from '{file_url}'")
traceback.print_exc()
return False
self._logger.info(f"Writing downloaded file content to disk: {output_path}") self._logger.info(f"Writing downloaded file content to disk: {output_path}")