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

el-garro · web-flow · commit 468ebf50cf59 · 2022-09-18T17:24:51.000+02:00
diff --git a/pyrogram/client.py b/pyrogram/client.py
@@ -737,6 +737,12 @@ async def handle_download(self, packet):
                 os.remove(file.name)
 
             return None
+        except asyncio.CancelledError:
+            if not in_memory:
+                file.close()
+                os.remove(file.name)
+                
+            raise asyncio.CancelledError()
         else:
             if in_memory:
                 file.name = file_name
