From b24cbc952ddef2b3d35c25cf6a83ab9e5dbd96dd Mon Sep 17 00:00:00 2001
From: yasirarism <55983182+yasirarism@users.noreply.github.com>
Date: Fri, 28 Apr 2023 13:03:30 +0700
Subject: [PATCH] Fix Msg id Invalid

---
 misskaty/plugins/imdb_search.py | 52 ++++++++++++++++++---------------
 1 file changed, 28 insertions(+), 24 deletions(-)

diff --git a/misskaty/plugins/imdb_search.py b/misskaty/plugins/imdb_search.py
index 5c895e8a..23ea3d45 100644
--- a/misskaty/plugins/imdb_search.py
+++ b/misskaty/plugins/imdb_search.py
@@ -155,12 +155,13 @@ async def imdb_search_id(kueri, message):
         )
         buttons.add(*BTN)
         msg = await k.edit_caption(msg, reply_markup=buttons)
-        await msg.wait_for_click(
-            from_user_id=message.from_user.id,
-            timeout=30
-        )
-    except ListenerTimeout:
-        await msg.edit_caption("😶‍🌫️ Waktu Habis. Task Telah Dibatalkan!")
+        try:
+            await msg.wait_for_click(
+                from_user_id=message.from_user.id,
+                timeout=30
+            )
+        except ListenerTimeout:
+            await msg.edit_caption("😶‍🌫️ Waktu Habis. Task Telah Dibatalkan!")
     except Exception as err:
         await k.edit_caption(f"Ooppss, gagal mendapatkan daftar judul di IMDb. Mungkin terkena rate limit atau down.\n\n<b>ERROR:</b> <code>{err}</code>")
 
@@ -211,12 +212,13 @@ async def imdb_search_en(kueri, message):
         )
         buttons.add(*BTN)
         msg = await k.edit_caption(msg, reply_markup=buttons)
-        await msg.wait_for_click(
-            from_user_id=message.from_user.id,
-            timeout=30
-        )
-    except ListenerTimeout:
-        await msg.edit_caption("😶‍🌫️ Timeout. Task Has Been Cancelled!")
+        try:
+            await msg.wait_for_click(
+                from_user_id=message.from_user.id,
+                timeout=30
+            )
+        except ListenerTimeout:
+            await msg.edit_caption("😶‍🌫️ Timeout. Task Has Been Cancelled!")
     except Exception as err:
         await k.edit_caption(f"Failed when requesting movies title. Maybe got rate limit or down.\n\n<b>ERROR:</b> <code>{err}</code>")
 
@@ -263,12 +265,13 @@ async def imdbcari(self: Client, query: CallbackQuery):
             )
             buttons.add(*BTN)
             msg = await query.message.edit_caption(msg, reply_markup=buttons)
-            await msg.wait_for_click(
-                from_user_id=int(uid),
-                timeout=30
-            )
-        except ListenerTimeout:
-            await msg.edit_caption("😶‍🌫️ Waktu Habis. Task Telah Dibatalkan!")
+            try:
+                await msg.wait_for_click(
+                    from_user_id=int(uid),
+                    timeout=30
+                )
+            except ListenerTimeout:
+                await msg.edit_caption("😶‍🌫️ Waktu Habis. Task Telah Dibatalkan!")
         except Exception as err:
             await query.message.edit_caption(f"Ooppss, gagal mendapatkan daftar judul di IMDb. Mungkin terkena rate limit atau down.\n\n<b>ERROR:</b> <code>{err}</code>")
     else:
@@ -308,12 +311,13 @@ async def imdbcari(self: Client, query: CallbackQuery):
             )
             buttons.add(*BTN)
             msg = await query.message.edit_caption(msg, reply_markup=buttons)
-            await msg.wait_for_click(
-                from_user_id=int(uid),
-                timeout=30
-            )
-        except ListenerTimeout:
-            await msg.edit_caption("😶‍🌫️ Timeout. Task Has Been Cancelled!")
+            try:
+                await msg.wait_for_click(
+                    from_user_id=int(uid),
+                    timeout=30
+                )
+            except ListenerTimeout:
+                await msg.edit_caption("😶‍🌫️ Timeout. Task Has Been Cancelled!")
         except Exception as err:
             await query.message.edit_caption(f"Failed when requesting movies title. Maybe got rate limit or down.\n\n<b>ERROR:</b> <code>{err}</code>")