floodwait with username

SpringsFern · May 10, 2024 · 97023a3 · 97023a3
1 parent b9f1b39
commit 97023a3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/WebStreamer/utils/custom_dl.py b/WebStreamer/utils/custom_dl.py
@@ -7,6 +7,7 @@
 from typing import Dict, Union
 from WebStreamer.bot import StreamBot, work_loads
 from pyrogram import Client, utils, raw
+from pyrogram.errors import FloodWait
 from .file_properties import get_file_ids
 from pyrogram.session import Session, Auth
 from pyrogram.errors import AuthBytesInvalid
@@ -189,6 +190,9 @@ async def yield_file(
 
         except (TimeoutError, AttributeError):
             pass
+        except FloodWait as e:
+            logging.error(f"{e} for client {client.username}")
+            raise e
         finally:
             work_loads[file_id.index] -= 1