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

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