fix: raise error instead of swallowing it in handle_download() (#310)

users choice best of 2026

Author: AnonymousProDev

pyrogram/client.py

@@ -1095,13 +1095,10 @@ async def handle_download(self, packet):
                 file.close()
                 os.remove(temp_file_path)
 
-            if isinstance(e, asyncio.CancelledError):
-                raise e
-
-            if isinstance(e, (FloodWait, FloodPremiumWait)):
-                raise e
+            if isinstance(e, pyrogram.StopTransmission):
+                return None
 
-            return None
+            raise e
         else:
             if in_memory:
                 file.name = file_name
@@ -1296,12 +1293,8 @@ def _check_all_hashes():
                         raise e
                     finally:
                         await cdn_session.stop()
-            except pyrogram.StopTransmission:
-                raise
-            except (FloodWait, FloodPremiumWait):
-                raise
             except Exception as e:
-                log.exception(e)
+                raise e
 
     async def get_session(
         self,