From f468909f3e52d375457dc13525c6c24509d8e7e9 Mon Sep 17 00:00:00 2001
From: Yasir Aris M <git@yasir.id>
Date: Fri, 10 Jan 2025 18:05:28 +0700
Subject: [PATCH] Add Error Handler

Saatnya back fix bugs

Signed-off-by: Yasir Aris M <git@yasir.id>
---
 misskaty/plugins/dev.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/misskaty/plugins/dev.py b/misskaty/plugins/dev.py
index aada9fd4..8990c0bc 100644
--- a/misskaty/plugins/dev.py
+++ b/misskaty/plugins/dev.py
@@ -38,6 +38,8 @@ from pyrogram.errors import (
     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):