From 48e32cf58313a2b333208934c8fa19ae9c900c87 Mon Sep 17 00:00:00 2001
From: being <34680324+being24@users.noreply.github.com>
Date: Tue, 30 Jan 2024 15:58:56 +0000
Subject: [PATCH] Refactor polling cog and fix delete_after bug

---
 cogs/polling.py | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/cogs/polling.py b/cogs/polling.py
index 86b28f9..18910b2 100644
--- a/cogs/polling.py
+++ b/cogs/polling.py
@@ -9,6 +9,7 @@
 from .utils.polling_manager import PollingManager, PollingParameter
 from .utils.setting_manager import SettingManager
 
+c = CommonUtil()
 logger = logging.getLogger("discord")
 polling_mng = PollingManager()
 
@@ -29,7 +30,6 @@ def __init__(self):
         super().__init__(
             placeholder="対象を選択してください", min_values=0, max_values=25
         )
-        # self.aggregation_mng = AggregationManager()
 
     async def callback(self, interaction: discord.Interaction):
         if self.view is None:
@@ -39,6 +39,9 @@ async def callback(self, interaction: discord.Interaction):
             item.disabled = True
         await self.view.message.edit(view=self.view)
 
+        if isinstance(interaction.message, discord.Message):
+            await c.delete_after(interaction.message)
+
         now = discord.utils.utcnow()
         poll_time_stamp = f"<t:{int(now.timestamp())}:f>"
 
@@ -91,6 +94,7 @@ async def on_timeout(self):
         try:
             await self.message.edit(view=self)  # type: ignore
         except discord.NotFound:
+            print("not found")
             pass
 
     async def interaction_check(self, interaction: discord.Interaction) -> bool:
@@ -126,7 +130,6 @@ class Polling(commands.Cog):
     def __init__(self, bot):
         self.bot = bot
         self.setting_mng = SettingManager()
-        self.c = CommonUtil()
 
         self.o = "\N{HEAVY LARGE CIRCLE}"
         self.x = "\N{CROSS MARK}"
@@ -153,15 +156,15 @@ async def poll(
         interaction: discord.Interaction,
         question: str,
         option_1: str,
-        option_2: str = None,
-        option_3: str = None,
-        option_4: str = None,
-        option_5: str = None,
-        option_6: str = None,
-        option_7: str = None,
-        option_8: str = None,
-        option_9: str = None,
-        option_10: str = None,
+        option_2: str | None = None,
+        option_3: str | None = None,
+        option_4: str | None = None,
+        option_5: str | None = None,
+        option_6: str | None = None,
+        option_7: str | None = None,
+        option_8: str | None = None,
+        option_9: str | None = None,
+        option_10: str | None = None,
     ) -> None:
         """optionで選択肢を追加します.ロールは最後に選択してください."""