salzarbeiter/src/ui.py

25 lines
812 B
Python

from discord import Interaction, ButtonStyle, PartialEmoji
from discord.ui import View
from emoji import emojize
from src.roles import AddRoleButton, RemoveRoleButton
class RoleSelectView(View):
def __init__(self):
super().__init__(timeout=None)
self.add_item(
AddRoleButton(
style=ButtonStyle.secondary,
label="Add Roles",
emoji=PartialEmoji.from_str(emojize(":check_mark_button:")),
custom_id="role_select:add"
)
)
self.add_item(
RemoveRoleButton(
style=ButtonStyle.secondary,
label="Remove Roles",
emoji=PartialEmoji.from_str(emojize(":cross_mark:")),
custom_id="role_select:remove"
)
)