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