forked from cswimr/SeaCogs
24 lines
1.1 KiB
Python
24 lines
1.1 KiB
Python
from discord import ui, Interaction
|
|
from redbot.core import commands
|
|
|
|
from aurora.configuration.embed import immune
|
|
from aurora.utilities.config import config
|
|
|
|
class Immune(ui.View):
|
|
def __init__(self, ctx: commands.Context):
|
|
super().__init__()
|
|
self.ctx = ctx
|
|
|
|
@ui.select(cls=ui.RoleSelect, placeholder="Select a role")
|
|
async def immune_select(self, interaction: Interaction, select: ui.RoleSelect,):
|
|
if not interaction.user.guild_permissions.manage_guild and not interaction.user.guild_permissions.administrator:
|
|
await interaction.response.send_message("You must have the manage guild permission to add immune roles.", ephemeral=True)
|
|
return
|
|
await interaction.response.defer()
|
|
immune_roles: list = await config.guild(self.ctx.guild).immune_roles()
|
|
if select.values[0].id in immune_roles:
|
|
immune_roles.remove(select.values[0].id)
|
|
else:
|
|
immune_roles.append(select.values[0].id)
|
|
await config.guild(self.ctx.guild).immune_roles.set(immune_roles)
|
|
await interaction.message.edit(embed=await immune(self.ctx), view=Immune(self.ctx))
|