Try to add markup

This commit is contained in:
2024-07-31 00:44:19 +03:00
parent a8f39d4630
commit d3ed6e4006
7 changed files with 106 additions and 9 deletions
+2 -2
View File
@@ -5,6 +5,6 @@ from .arguments import ArgumentsMiddleware
from .database import DatabaseMiddleware
def setup_middlewares(bot: TeleBot, i18n: I18N, engine):
bot.setup_middleware(ArgumentsMiddleware(i18n))
def setup_middlewares(bot: TeleBot, i18n: I18N, engine, markup):
bot.setup_middleware(ArgumentsMiddleware(i18n, markup))
bot.setup_middleware(DatabaseMiddleware(engine))
+11 -3
View File
@@ -3,16 +3,24 @@ from telebot.types import Message, CallbackQuery
class ArgumentsMiddleware (BaseMiddleware):
def __init__(self, i18n):
def __init__(self, i18n, markup):
super().__init__()
self.i18n = i18n
self.markup = markup
self.update_types = ["message", "callback_query"]
def pre_process(self, obj, data: dict):
if isinstance(obj, Message):
data["t"] = self.i18n.customized_call(message=obj)
self.pre_process_message(obj, data)
elif isinstance(obj, CallbackQuery):
data["t"] = self.i18n.customized_call(callback=obj)
self.pre_process_callback(obj, data)
data["m"] = self.markup
def pre_process_message(self, message: Message, data: dict):
data["t"] = self.i18n.customized_call(message=message)
def pre_process_callback(self, call: CallbackQuery, data: dict):
data["t"] = self.i18n.customized_call(callback=call)
def post_process(self, message, data: dict, exception: BaseException):
pass