Delete message after user selection

This is to prevent users from sending too many messages.
MasterJ93 · Nov 27, 2023 · ab424e0 · ab424e0
1 parent c4d1ee6
commit ab424e0
Showing 1 changed file with 18 additions and 5 deletions.
diff --git a/views/beginner_views.py b/views/beginner_views.py
@@ -1,6 +1,7 @@
 """
 Contains views related to members that just came into the server.
 """
+import asyncio
 import discord
 
 from config import settings
@@ -61,7 +62,7 @@ async def selection_1(self,
                 view=view
             )
 
-        await self.send_message_and_end_onboarding(interaction.response,
+        await self.send_message_and_end_onboarding(interaction,
                                                    response_message)
 
     @discord.ui.button(
@@ -81,7 +82,7 @@ async def selection_2(self,
                     billboards=self.alliance_billboard.mention
         )
 
-        await self.send_message_and_end_onboarding(interaction.response,
+        await self.send_message_and_end_onboarding(interaction,
                                                    response_message)
 
     @discord.ui.button(
@@ -99,25 +100,37 @@ async def selection_3(self,
 
         response_message = beginner['OPTION_3_RESPONSE']
 
-        await self.send_message_and_end_onboarding(interaction.response,
+        await self.send_message_and_end_onboarding(interaction,
                                                    response_message)
 
     async def send_message_and_end_onboarding(
         self,
-        response: discord.InteractionResponse,
+        interaction: discord.Interaction,
         response_message: str):
         """
         Sends an ephemeral message, then finds and deletes the
         original message.
         """
 
         # Ephemeral message is sent, then is deleted after a minute.
-        await response.send_message(
+        await interaction.response.send_message(
             content=response_message,
             ephemeral=True,
             delete_after=60.0
         )
 
+        message_counter = 3
+        async for message in \
+            interaction.channel.history(limit=200): #type: ignore
+            if message.author.bot is True:
+                await message.delete()
+                await asyncio.sleep(3)
+                message_counter -= 1
+
+                if message_counter == 0:
+                    break
+
+
 class ClanInviteInterestView(discord.ui.View):
     """
     Contains a view for admins to decide whether to