27 lines
929 B
Python
27 lines
929 B
Python
from telebot.handler_backends import BaseMiddleware
|
|
from telebot.types import Message, CallbackQuery
|
|
|
|
|
|
class ArgumentsMiddleware (BaseMiddleware):
|
|
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):
|
|
self.pre_process_message(obj, data)
|
|
elif isinstance(obj, CallbackQuery):
|
|
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
|