mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-29 05:18:10 +00:00
Code Issues Releases Wiki Activity

Check whether get_file failed or not #37

Browse Source 
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
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pyrogram/client/client.py
View File

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