Return button for promote and infract and smth el

bugsbirb · Feb 20, 2024 · 6650ecb · 6650ecb
1 parent e656e57
commit 6650ecb
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/Cogs/Configuration/Configuration.py b/Cogs/Configuration/Configuration.py
@@ -119,7 +119,10 @@ async def callback(self, interaction: discord.Interaction):
             else:
                 scollection.insert_one(data)
             await refreshembed(interaction)    
-            await interaction.response.edit_message(content=None)
+            try:
+             await interaction.response.edit_message(content=None)
+            except:
+                print('Error editing staff role config message') 
         except Exception as e:
             print(f"An error occurred: {str(e)}")
         print(f"Select Roles {selected_role_ids}")
@@ -143,7 +146,10 @@ async def callback(self, interaction: discord.Interaction):
 
         arole.update_one({'guild_id': interaction.guild.id}, {'$set': data}, upsert=True)
         await refreshembed(interaction)  
-        await interaction.response.edit_message(content=None)
+        try:
+         await interaction.response.edit_message(content=None)
+        except: 
+            print('Error editing admin role config message') 
         print(f"Select Roles {selected_role_ids}")
 
 
@@ -837,5 +843,6 @@ async def refreshembed(interaction):
             embed.set_author(name=interaction.guild.name, icon_url=interaction.guild.icon)
             await interaction.message.edit(embed=embed)
 
+
 async def setup(client: commands.Bot) -> None:
     await client.add_cog(ConfigCog(client))     
diff --git a/Cogs/Modules/adminpanel.py b/Cogs/Modules/adminpanel.py
@@ -884,9 +884,10 @@ async def Promote(
             )
             return
 
-        view = PromotionRoleView(self.user, self.guild, self.author)
+        view = Return(self.user, self.guild, self.author)
+        view.add_item(PromotionRole(self.user, self.guild, self.author))
         await interaction.response.edit_message(view=view)
-
+        
     @discord.ui.button(
         label="Infract",
         style=discord.ButtonStyle.grey,
@@ -915,7 +916,8 @@ async def Infract(
             )
             return
 
-        view = InfractionOptionView(self.user, self.guild, self.author)
+        view = Return(self.user, self.guild, self.author)
+        view.add_item(InfractionOption(self.user, self.guild, self.author))
         await interaction.response.edit_message(view=view)
 
     @discord.ui.button(
@@ -1084,8 +1086,9 @@ def __init__(self, user, guild, author):
         label="Return",
         style=discord.ButtonStyle.grey,
         emoji="<:Return:1166514220960063568>",
+        row=2
     )
-    async def Return(self, interaction: discord.Interaction, button: discord.ui.Button):
+    async def Return77(self, interaction: discord.Interaction, button: discord.ui.Button):
         if interaction.user.id != self.author.id:
             embed = discord.Embed(
                 description=f"**{interaction.user.global_name},** this is not your view!",