machka-pasla
3d58f60a4d
- Integrated new service building functions to streamline bot initialization, enhancing maintainability and clarity. - Updated the dispatcher setup to include core services dynamically, reducing redundancy in service registration. - Added support for multiple broadcast targets in the admin handler, allowing for more flexible message distribution. - Enhanced localization for new broadcast target options, improving user experience in both English and Russian. - Implemented new database access methods to retrieve user IDs based on subscription status, optimizing broadcast logic.
42 lines
1.7 KiB
Python
42 lines
1.7 KiB
Python
import logging
|
|
from typing import Dict
|
|
|
|
from aiogram import Bot, Dispatcher
|
|
from aiogram.enums import ParseMode
|
|
from aiogram.client.default import DefaultBotProperties
|
|
from aiogram.fsm.storage.memory import MemoryStorage
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from config.settings import Settings
|
|
from bot.middlewares.db_session import DBSessionMiddleware
|
|
from bot.middlewares.i18n import I18nMiddleware, get_i18n_instance, JsonI18n
|
|
from bot.middlewares.ban_check_middleware import BanCheckMiddleware
|
|
from bot.middlewares.action_logger_middleware import ActionLoggerMiddleware
|
|
from bot.middlewares.profile_sync import ProfileSyncMiddleware
|
|
from bot.routers import build_root_router
|
|
|
|
|
|
def build_dispatcher(settings: Settings, async_session_factory: sessionmaker) -> tuple[Dispatcher, Bot, Dict]:
|
|
storage = MemoryStorage()
|
|
default_props = DefaultBotProperties(parse_mode=ParseMode.HTML)
|
|
bot = Bot(token=settings.BOT_TOKEN, default=default_props)
|
|
|
|
dp = Dispatcher(storage=storage, settings=settings, bot_instance=bot)
|
|
|
|
i18n_instance = get_i18n_instance(path="locales", default=settings.DEFAULT_LANGUAGE)
|
|
|
|
dp["i18n_instance"] = i18n_instance
|
|
dp["async_session_factory"] = async_session_factory
|
|
|
|
dp.update.outer_middleware(DBSessionMiddleware(async_session_factory))
|
|
dp.update.outer_middleware(I18nMiddleware(i18n=i18n_instance, settings=settings))
|
|
dp.update.outer_middleware(ProfileSyncMiddleware())
|
|
dp.update.outer_middleware(BanCheckMiddleware(settings=settings, i18n_instance=i18n_instance))
|
|
dp.update.outer_middleware(ActionLoggerMiddleware(settings=settings))
|
|
|
|
dp.include_router(build_root_router(settings))
|
|
|
|
return dp, bot, {"i18n_instance": i18n_instance}
|
|
|
|
|