Template
Try to add markup
This commit is contained in:
@@ -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))
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user