FunkyBot/views/LanguageButtons.py

20 lines
488 B
Python
Raw Normal View History

import discord
from libs.View import View
from data.models.Language import Language
from libs.LanguageButton import LanguageButton
class LanguageButtons(View):
def __init__(self, bot):
self.bot = bot
super().__init__()
def get_buttons(self):
languages = self.bot.db.session.query(Language).all()
for language in languages:
btn = LanguageButton(label=language.label)
self.buttons.append(btn)
return self.buttons