Marc Ahlgrim
74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
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__)
|