Refactored config

This commit is contained in:
2024-07-26 02:58:19 +03:00
parent 7b0aac673f
commit 273c0bf2da
10 changed files with 144 additions and 85 deletions
+3 -4
View File
@@ -1,9 +1,8 @@
import logging
from telebot import TeleBot
from .arguments import ExtraArguments
from ..i18n import I18N
def setup_middlewares(bot: TeleBot, logger: logging.Logger, i18n):
bot.setup_middleware(ExtraArguments(logger, i18n))
def setup_middlewares(bot: TeleBot, i18n: I18N):
bot.setup_middleware(ExtraArguments(i18n))
+1 -5
View File
@@ -1,18 +1,14 @@
import logging
from telebot.handler_backends import BaseMiddleware
from telebot.types import Message, CallbackQuery
class ExtraArguments(BaseMiddleware):
def __init__(self, logger: logging.Logger, i18n):
def __init__(self, i18n):
super().__init__()
self.logger = logger
self.i18n = i18n
self.update_types = ["message", "callback_query"]
def pre_process(self, obj, data: dict):
data["logger"] = self.logger
if isinstance(obj, Message):
data["t"] = self.i18n.customized_call(message=obj)
elif isinstance(obj, CallbackQuery):