SeaCogs/aurora/configuration/menus/addrole.py
SeaswimmerTheFsh 464fa9f97a
Some checks failed
Actions / Lint Code (Pylint) (pull_request) Failing after 14s
Actions / Build Documentation (MkDocs) (pull_request) Successful in 12s
fix(aurora): fixed some errors
2024-01-16 13:06:50 +00:00

24 lines
1.2 KiB
Python

from discord import ui, Interaction
from redbot.core import commands
from aurora.configuration.embed import addrole
from aurora.utilities.config import config
class Addrole(ui.View):
def __init__(self, ctx: commands.Context):
super().__init__()
self.ctx = ctx
@ui.select(cls=ui.RoleSelect, placeholder="Select a role")
async def addrole_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 roles to the addrole whitelist.", ephemeral=True)
return
await interaction.response.defer()
addrole_whitelist: list = await config.guild(self.ctx.guild).addrole_whitelist()
if select.values[0].id in addrole_whitelist:
addrole_whitelist.remove(select.values[0].id)
else:
addrole_whitelist.append(select.values[0].id)
await config.guild(self.ctx.guild).addrole_whitelist.set(addrole_whitelist)
await interaction.message.edit(embed=await addrole(self.ctx), view=Addrole(self.ctx))