spalencsar/bearhub
1
0
Fork 0
mirror of https://github.com/spalencsar/bearhub.git synced 2026-07-06 20:34:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Vinicius Moreira
2023-12-14 14:17:57 -03:00
parent e44db92a1d
commit 0bc66a9d16
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
bauh/view/core/downloader.py
View File

@@ -73,11 +73,16 @@ class SelfFileDownloader(FileDownloader):
known_size = content_length and content_length > 0
total_size_str = get_human_size_str(content_length) if known_size > 0 else "?"
for data in res.iter_content(chunk_size=1024):
byte_stream.write(data)
total_downloaded += len(data)
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})")
try:
for data in res.iter_content(chunk_size=1024):
byte_stream.write(data)
total_downloaded += len(data)
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}")