From d69f43ed72c0c3ff8b82b28d009df7c78d5c309d Mon Sep 17 00:00:00 2001
From: npc203 <samjerub@gmail.com>
Date: Sat, 12 Feb 2022 20:22:15 +0530
Subject: [PATCH] [SlashTags] Fix buttons when skipped dynamically

---
 slashtags/testing/button_menus.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/slashtags/testing/button_menus.py b/slashtags/testing/button_menus.py
index f3c97876..9f0c6401 100644
--- a/slashtags/testing/button_menus.py
+++ b/slashtags/testing/button_menus.py
@@ -194,7 +194,7 @@ async def _internal_loop(self):
                     raise asyncio.TimeoutError()
 
                 # Exception will propagate if e.g. cancelled or timed out
-                payload = done.pop().result()
+                payload: InteractionButton = done.pop().result()
                 loop.create_task(self.update(payload))
 
                 # NOTE: Removing the reaction ourselves after it's been done when
@@ -250,6 +250,8 @@ async def update(self, payload: InteractionButton):
             return
         if not self._running:
             return
+        if not button.skip_if(self):
+            return
 
         try:
             if button.lock: