yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-02 05:24:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Make get_file and save_file handle StopTransmission errors

Browse source
This commit is contained in:
Dan 2019-01-11 14:03:16 +01:00
parent 6b63e88de7
commit b37d4dc7ec
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
pyrogram/client/client.py
View file

@ -1368,6 +1368,8 @@ class Client(Methods, BaseClient):
if progress:
progress(self, min(file_part * part_size, file_size), file_size, *progress_args)
except Client.StopTransmission:
raise
except Exception as e:
log.error(e, exc_info=True)
else:
@ -1569,7 +1571,8 @@ class Client(Methods, BaseClient):
except Exception as e:
raise e
except Exception as e:
log.error(e, exc_info=True)
if not isinstance(e, Client.StopTransmission):
log.error(e, exc_info=True)
try:
os.remove(file_name)