Added poll_check to commands.py

SylteA · Oct 26, 2023 · de91f3b · de91f3b
1 parent 0062559
commit de91f3b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/bot/extensions/polls/commands.py b/bot/extensions/polls/commands.py
@@ -24,6 +24,15 @@ def reactions(self):
             9: "9️⃣",
             10: "🔟",
         }
+
+    def poll_check(self, message: discord.Message):
+        try:
+            embed = message.embeds[0]
+        except Exception:
+            return False
+        if str(embed.footer.text).count("Poll by") == 1:
+            return message.author == self.bot.user
+        return False
 
     @app_commands.command()
     @app_commands.checks.cooldown(1, 10)