from bot.Handler import Handler
from bot import SET_DATE_STEP, SET_TABLE_STEP
from bot.keyboards.DateToBookingKeyboard import DateToBookingKeyboard
from bot.keyboards.GetTableListKeyboard import GetTableListKeyboard


class BookingHandler(Handler):

    @staticmethod
    def booking_vip(update, context):
        query = update.callback_query
        query.answer()

        reply_markup = DateToBookingKeyboard()
        query.edit_message_text(
            'Выберите удобную дату:', reply_markup=reply_markup.create_keyboard()
        )

        return SET_DATE_STEP

    @staticmethod
    def booking_table(update, context):
        query = update.callback_query
        query.answer()

        reply_markup = GetTableListKeyboard()
        query.edit_message_text(
            'Выберите удобный стол:', reply_markup=reply_markup.create_keyboard()
        )

        return SET_TABLE_STEP