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

Add tempfile deletion in case of asyncio task cancellation. (#1080)

Browse Source
This commit is contained in:
Nyan 2022-09-18 11:24:51 -04:00 committed by GitHub
parent 81895d74c7
commit 468ebf50cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pyrogram/client.py
View File

@ -737,6 +737,12 @@ class Client(Methods):
os.remove(file.name) os.remove(file.name)
return None return None
except asyncio.CancelledError:
if not in_memory:
file.close()
os.remove(file.name)
raise asyncio.CancelledError()
else: else:
if in_memory: if in_memory:
file.name = file_name file.name = file_name