Release 0.3

Signed-off-by: Marc Ahlgrim <marc@onemarcfifty.com>
2022-07-22 12:50:06 +02:00
parent e58ba3a745
commit 6e35d9e064
15 changed files with 497 additions and 173 deletions
@@ -0,0 +1,73 @@
+import discord
+import traceback
+import config
+
+# ############################################
+# the Subscribe() class is a modal ui dialog
+# that let's you select one or multiple
+# roles that will be assigned to you.
+# The roles are set in the config.AUTO_EVENTS
+# variable
+# ############################################
+
+class Subscribe(discord.ui.Modal, title='(un)subscribe to Event-Notification'):
+    
+    # We need to make sure that the config is read at object definition time
+    # because AUTO_EVENTS might be empty otherwise
+    if config.AUTO_EVENTS == []:
+        config.readConfig()
+
+    # define the menu options (label=text, vaue=role_id)
+
+    menu_options = []
+    for menu_option in config.AUTO_EVENTS:
+        menu_options.append(discord.SelectOption(label= menu_option["notify_hint"],
+                                                 value=menu_option["subscription_role_num"]))
+    # define the UI dropdown menu Element
+
+    Menu = discord.ui.Select(
+        options = menu_options,
+        max_values=len(config.AUTO_EVENTS),
+        min_values=0)
+    
+
+    # #####################################
+    # on_submit is called when the user submits
+    # the modal. It will then go through
+    # the menu options and revoke / assign
+    # the corresponding roles
+    # #####################################
+    
+
+    async def on_submit(self, interaction: discord.Interaction):
+        role: discord.Role
+        roles = interaction.user.guild.roles
+        member: discord.Member
+        member = interaction.user
+
+        # first we remove all roles
+
+        for option in self.Menu.options:
+            role=member.get_role(option.value)
+            if not (role is None):
+                await member.remove_roles(role)
+
+        # then we assign all selected roles
+        # unfortunately we need to loop through all rolles
+        # maybe there is a better solution ?
+
+        for option in self.Menu._selected_values:
+            for role in roles:
+                if int(option) == int(role.id):
+                    await member.add_roles(role)
+
+
+        await interaction.response.send_message(f'Thanks for using my services !', ephemeral=True)
+
+                                                            
+
+    async def on_error(self, interaction: discord.Interaction, error: Exception) -> None:
+        await interaction.response.send_message('Oops! Something went wrong.', ephemeral=True)
+
+        # Make sure we know what the error actually is
+        traceback.print_tb(error.__traceback__)