Add Error Handler

Saatnya back fix bugs

Signed-off-by: Yasir Aris M <[email protected]>

misskaty/plugins/dev.py

@@ -38,6 +38,8 @@
     FloodWait,
     MessageTooLong,
     PeerIdInvalid,
+    RPCError,
+    SlowmodeWait,
 )
 from pyrogram.raw.types import UpdateBotStopped
 from pyrogram.types import (
@@ -659,6 +661,17 @@ async def update_restart(_, ctx: Message, strings):
     os.execvp(sys.executable, [sys.executable, "-m", "misskaty"])
 
 
+@app.on_error(errors=(Exception, FloodWait, RPCError, SlowmodeWait))
+async def error_handlers(_: "Client", __: "Update", error: "Exception") -> None:
+    if isinstance(error, (FloodWait, SlowmodeWait)):
+        await asyncio.sleep(error.value)
+    # else:
+        # if config.DEBUG_MODE:
+        #    LOGGER.error(repr(error))
+        # else:
+        #    return None
+
+
 @app.on_raw_update(group=-99)
 async def updtebot(client, update, users, _):
     if isinstance(update, UpdateBotStopped):