Fix feedback button bugs

TLNBS2405 · Feb 18, 2024 · 6504169 · 6504169
1 parent 7467cdb
commit 6504169
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/main.py b/src/main.py
@@ -33,6 +33,8 @@ async def on_ready(self):
         if not self.synced:
             await tree.sync()
             self.synced = True
+        action_channel = self.get_channel(actioned_channel_id)
+        self.add_view(view=button.DoneButton(action_channel))
         print('Logged on as', self.user)
 
 

diff --git a/src/module/button.py b/src/module/button.py
@@ -1,19 +1,23 @@
+import random
+
+import sys
+
 import discord
 
 
 class DoneButton(discord.ui.View):
     def __init__(self, actioned_channel):
         self.actioned_channel = actioned_channel
-        super().__init__()
+        super().__init__(timeout=None)
 
-    @discord.ui.button(label="👍", style=discord.ButtonStyle.green)
+    @discord.ui.button(label="👍", style=discord.ButtonStyle.green, custom_id="Done")
     async def done(self, interaction: discord.Interaction, button: discord.ui.Button):
         done_message = "{} \nactioned by **{}** with 👍\n".format(
             interaction.message.content, interaction.user.name)
         await self.actioned_channel.send(content=done_message)
         await interaction.message.delete()
 
-    @discord.ui.button(label="👎", style=discord.ButtonStyle.red)
+    @discord.ui.button(label="👎", style=discord.ButtonStyle.red, custom_id="Cancel")
     async def cancel(self, interaction: discord.Interaction, button: discord.ui.Button):
         done_message = "{} \nactioned by **{}** with 👎\n".format(
             interaction.message.content, interaction.user.name)