2
0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 12:57:52 +00:00

Check whether get_file failed or not #37

If it failed, also delete any eventual temporary file
This commit is contained in:
Dan 2018-03-20 19:25:23 +01:00
parent 440654a63f
commit 2b33f23990

View File

@ -567,6 +567,9 @@ class Client:
progress=progress
)
if tmp_file_name is None:
return None
if file_name is not None:
path[0] = "downloads/{}".format(file_name)
@ -2334,11 +2337,16 @@ class Client:
if len(chunk) < limit:
break
except Exception as e:
log.error(e)
raise e
finally:
cdn_session.stop()
except Exception as e:
log.error(e)
try:
os.remove(file_name)
except OSError:
pass
else:
return file_name
finally: