test roles

app.py

@@ -18,6 +18,8 @@ import asyncio # check if used
 import logging
 import urllib.parse
 
+from discord.ui import Button, View
+
 zurich_tz = timezone("Europe/Zurich")
 
 def convert_to_timezone(dt, tz):
@@ -541,6 +543,65 @@ async def on_voice_state_update(member, before, after):
     except Exception as e:
         print(f"on_voice_state_update Error: {e}")  
 
+# -------------------------------------------------------------------------------------------------------------------------------------
+class RoleToggleButton(Button):
+    def __init__(self, role: discord.Role):
+        # Set the button label to the role name and use a primary style.
+        super().__init__(label=role.name, style=discord.ButtonStyle.primary)
+        self.role = role
+
+    async def callback(self, interaction: discord.Interaction):
+        # Check if the user already has the role
+        if self.role in interaction.user.roles:
+            try:
+                await interaction.user.remove_roles(self.role)
+                await interaction.response.send_message(
+                    f"Removed role: {self.role.name}", ephemeral=True
+                )
+            except Exception as e:
+                await interaction.response.send_message(
+                    f"Error removing role: {e}", ephemeral=True
+                )
+        else:
+            try:
+                await interaction.user.add_roles(self.role)
+                await interaction.response.send_message(
+                    f"Added role: {self.role.name}", ephemeral=True
+                )
+            except Exception as e:
+                await interaction.response.send_message(
+                    f"Error adding role: {e}", ephemeral=True
+                )
+
+# A view that holds one toggle button per role.
+class RoleSelectionView(View):
+    def __init__(self, roles: list):
+        # Setting timeout to None allows the view to persist indefinitely.
+        super().__init__(timeout=None)
+        for role in roles:
+            self.add_item(RoleToggleButton(role))
+
+# Command that sends the role buttons message.
+@bot.command(name="role_buttons")
+async def role_buttons(ctx):
+    # Only allow lunarflu (ID 811235357663297546) to use this command.
+    if ctx.author.id != 811235357663297546:
+        await ctx.send("You are not authorized to use this command.", delete_after=10)
+        return
+
+    # List the role IDs you want to include (replace these with your actual role IDs).
+    role_ids = [1014517792550166630, 1014548568238997616, 1014548769355862036, 1077250031180071023, 1093982736961785877]
+    # Retrieve the Role objects from the guild.
+    roles = [ctx.guild.get_role(rid) for rid in role_ids if ctx.guild.get_role(rid) is not None]
+
+    # Create the view with buttons for each role.
+    view = RoleSelectionView(roles)
+    # Send a message with the view attached.
+    await ctx.send("Click the buttons below to toggle roles:", view=view)
+
+
+
+
 # github test stuff -------------------------------------------------------------------------------------------------------------------
 """
 async def check_github():