Re-raise asyncio.CancelledError to avoid continuations

delivrance · delivrance · commit 71f263b30d3a · 2022-09-18T17:55:50.000+02:00
diff --git a/pyrogram/client.py b/pyrogram/client.py
@@ -730,15 +730,17 @@ async def handle_download(self, packet):
         temp_file_path = os.path.abspath(re.sub("\\\\", "/", os.path.join(directory, file_name))) + ".temp"
         file = BytesIO() if in_memory else open(temp_file_path, "wb")
 
-        # noinspection PyBroadException
         try:
             async for chunk in self.get_file(file_id, file_size, 0, 0, progress, progress_args):
                 file.write(chunk)
-        except BaseException:
+        except BaseException as e:
             if not in_memory:
                 file.close()
                 os.remove(temp_file_path)
 
+            if isinstance(e, asyncio.CancelledError):
+                raise e
+
             return None
         else:
             if in_memory:
