2
0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-31 06:16:06 +00:00

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

This commit is contained in:
Nyan
2022-09-18 11:24:51 -04:00
committed by GitHub
parent 81895d74c7
commit 468ebf50cf

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