From cc1689ee575b9e6de0ba287f01a8aadc02822876 Mon Sep 17 00:00:00 2001 From: yasir Date: Mon, 9 Jan 2023 13:03:07 +0700 Subject: [PATCH] Rewrite IMDB Module --- misskaty/core/decorator/errors.py | 4 ++-- misskaty/plugins/imdb_search.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/misskaty/core/decorator/errors.py b/misskaty/core/decorator/errors.py index 4733aa76..aa693ac6 100644 --- a/misskaty/core/decorator/errors.py +++ b/misskaty/core/decorator/errors.py @@ -56,8 +56,8 @@ def capture_err(func): try: await app.send_message(LOG_CHANNEL, x) await message.reply(x) - except FloodWait: - await asyncio.sleep(x.value) + except FloodWait as e: + await asyncio.sleep(e.value) raise err return capture diff --git a/misskaty/plugins/imdb_search.py b/misskaty/plugins/imdb_search.py index 9c5e4a45..856b9c03 100644 --- a/misskaty/plugins/imdb_search.py +++ b/misskaty/plugins/imdb_search.py @@ -1,4 +1,5 @@ from logging import getLogger +from utils import demoji from pykeyboard import InlineKeyboard from pyrogram.types import ( InlineKeyboardMarkup,