import discord from libs.Roles import Roles from views.items.MatrixPenguinsButton import MatrixPenguinsButton from views.items.MatrixRefuserPenguinsButton import MatrixRefuserPenguinsButton class MatrixButtons(discord.ui.View): def __init__(self): super().__init__(timeout=None) self.matrix_role = None self.matrix_refuser_role = None self.add_item(MatrixPenguinsButton()) self.add_item(MatrixRefuserPenguinsButton()) # async def set_roles(self, guild): # self.matrix_role = await Roles().get_role(guild, "Matrix Penguins") # self.matrix_refuser_role = await Roles().get_role(guild, "Matrix-Refuser Penguins") # # @discord.ui.button(label="Matrix Penguins", style=discord.ButtonStyle.blurple) # async def matrix_click(self, interaction: discord.Interaction, button): # member = interaction.user # content = [] # await self.set_roles(interaction.guild) # # if self.matrix_role in member.roles: # if await Roles().remove_role(interaction.guild, "Matrix Penguins", member): # content.append("Matrix Penguins has been removed") # else: # if await Roles().remove_role(interaction.guild, "Matrix-Refuser Penguins", member): # content.append("Matrix-Refuser Penguins has been removed") # if await Roles().add_role(interaction.guild, "Matrix Penguins", member): # content.append("Matrix Penguins has been added") # # await interaction.response.send_message(" and ".join(content), delete_after=5, ephemeral=True) # # @discord.ui.button(label="Matrix-Refuser Penguin", style=discord.ButtonStyle.red) # async def refuser_click(self, interaction: discord.Interaction, button): # member = interaction.user # content = [] # await self.set_roles(interaction.guild) # # if self.matrix_refuser_role in member.roles: # if await Roles().remove_role(interaction.guild, "Matrix-Refuser Penguins", member): # content.append("Matrix-Refuser Penguin has been removed") # else: # if await Roles().remove_role(interaction.guild, "Matrix Penguins", member): # content.append("Matrix Penguins has been removed") # if await Roles().add_role(interaction.guild, "Matrix-Refuser Penguins", member): # content.append("Matrix-Refuser Penguins has been added") # # await interaction.response.send_message(" and ".join(content), delete_after=5, ephemeral=True)