from telegram import Update
from telegram.ext import CallbackContext, CallbackQueryHandler

from callbackQuery.data import data
from callbackQuery.edit import edit
from callbackQuery.new import new
from commands.start import start


def callback_query(update: Update, context: CallbackContext):
    if update.callback_query.data == "home":
        start(update, context)
    elif update.callback_query.data == "new":
        new(update, context)
    elif update.callback_query.data == "data":
        data(update, context)
    elif update.callback_query.data.startswith("edit_"):
        edit(update, context)


callback_query_handler = CallbackQueryHandler(callback_query)