from bot.Keyboard import Keyboard
from telegram import InlineKeyboardButton, InlineKeyboardMarkup


class GetTableListKeyboard(Keyboard):

    def get_keyboard(self):
        keyboard = []
        for i in range(1, 8):
            keyboard.append(
                [
                    InlineKeyboardButton("Стол {t}".format(t=i),
                                         callback_data="set_table={t}".format(t=i))
                ]
            )

        keyboard.append(
            [
                InlineKeyboardButton("Назад", callback_data="set_table=back")
            ]
        )

        return keyboard