funkywaddle/FunkyBot
funkywaddle/FunkyBot
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Update Framework and Language Button libs to use current methods

Browse source
This commit is contained in:
Funky Waddle 2025-06-29 02:20:52 -05:00
parent e018090af6
commit e6cf2194b2
2 changed files with 17 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
libs/FrameworkButton.py
View file

@ -12,8 +12,8 @@ class FrameworkButton(discord.ui.Button):
self.language_role = language_role
async def callback(self, interaction: discord.Interaction):
role = await Roles().get_role(interaction.guild, self.label)
language_role = await Roles().get_role(interaction.guild, self.language_role)
role = await Roles().get_guild_role(interaction.guild, self.label)
language_role = await Roles().get_guild_role(interaction.guild, self.language_role)
messages = []
if role in interaction.user.roles:
await Roles().remove_role(interaction.guild, self.label, interaction.user)

27
libs/LanguageButton.py
View file

@ -1,33 +1,36 @@
import discord
import importlib
import os
from discord.ui import Button
from discord import Interaction, ButtonStyle
from libs.Roles import Roles
from data.models.Language import Language
class LanguageButton(discord.ui.Button):
class LanguageButton(Button):
def __init__(self, style=discord.ButtonStyle.blurple, label="Not Set"):
def __init__(self, style=ButtonStyle.blurple, label="Not Set"):
super().__init__()
self.style = style
self.label = label
self.disabled: bool = False
async def callback(self, interaction: discord.Interaction):
role = await Roles().get_role(interaction.guild, self.label)
async def callback(self, interaction: Interaction):
role = await Roles().get_guild_role(interaction.guild, self.label)
language = interaction.client.db.session.query(Language).filter(Language.label == self.label).first()
lang_frameworks = [fw.label for fw in language.frameworks]
framework_roles = await Roles().get_roles(interaction.guild, lang_frameworks)
frameworks_to_remove = list(set(framework_roles) & set(interaction.user.roles))
messages = []
if role in interaction.user.roles:
await Roles().remove_role(interaction.guild, self.label, interaction.user)
lang_frameworks = await self.bot.db.language_frameworks(interaction.client.db)
framework_roles = await Roles().get_roles(interaction.guild, lang_frameworks)
frameworks_to_remove = list(set(framework_roles) & set(interaction.user.roles))
for framework in frameworks_to_remove:
await Roles().remove_role(interaction.guild, framework.name, interaction.user)
await Roles().remove_role(framework, interaction.user)
messages.append(f"You are no longer associated with the {framework.name} framework group")
messages.append(f"You are no longer associated with the {self.label} language group")
else:
await Roles().add_role(interaction.guild, self.label, interaction.user)
messages.append(f"You are now associated with the {self.label} language group")
await interaction.response.send_message("\n".join(messages), delete_after=5, ephemeral=True)
async def get_language_frameworks(self, db):
language = db.session.query(Language).filter(Language.label == self.label).first()
lang_frameworks = [fw.label for fw in language.frameworks]
return lang_frameworks