bot.py

import asyncio
import logging

from aiogram import Bot, Dispatcher
from aiogram.dispatcher.fsm.storage.redis import RedisStorage, DefaultKeyBuilder

from migrations.from_redis_requests_to_postgres import migrate_requests
from schedulers.base import setup_scheduler
from tgbot.config import load_config
from tgbot.handlers.admin import admin_router
from tgbot.handlers.edit_request import edit_request_router
from tgbot.handlers.evacuation import evacuation_router
from tgbot.handlers.get_medicine import medicine_router
from tgbot.handlers.provision import provision_router
from tgbot.handlers.search_people import search_people_router
from tgbot.handlers.user import user_router
from tgbot.infrastructure.database.functions.setup import create_session_pool
from tgbot.middlewares.config import ConfigMiddleware
from tgbot.middlewares.database import DatabaseMiddleware
from tgbot.services import broadcaster
from tgbot.services.default_commands import set_default_commands

logger = logging.getLogger(__name__)


async def on_startup(bot: Bot, admin_ids: list[int]):
    await broadcaster.broadcast(bot, admin_ids, "Бот був запущений")
    await set_default_commands(bot)


def register_global_middlewares(dp: Dispatcher, config, session_pool):
    dp.message.outer_middleware(ConfigMiddleware(config))
    dp.callback_query.outer_middleware(ConfigMiddleware(config))

    dp.message.outer_middleware(DatabaseMiddleware(session_pool))
    dp.callback_query.outer_middleware(DatabaseMiddleware(session_pool))


async def main():
    logging.basicConfig(
        level=logging.INFO,
        format=u'%(filename)s:%(lineno)d #%(levelname)-8s [%(asctime)s] - %(name)s - %(message)s',
    )
    logger.info("Starting bot")
    config = load_config(".env")

    storage = RedisStorage.from_url(config.redis_config.dsn(), key_builder=DefaultKeyBuilder(with_bot_id=True))
    bot = Bot(token=config.tg_bot.token, parse_mode='HTML')
    dp = Dispatcher(storage=storage)
    session_pool = await create_session_pool(config.db, echo=False)

    scheduler = setup_scheduler(bot, config, storage, session_pool)

    dp['config'] = config
    dp['scheduler'] = scheduler

    await migrate_requests(session_pool, storage)
    for router in [
        admin_router,
        edit_request_router,
        user_router,
        evacuation_router,
        medicine_router,
        # provision_router,
        search_people_router,
    ]:
        dp.include_router(router)

    register_global_middlewares(dp, config, session_pool)

    scheduler.start()
    await on_startup(bot, config.tg_bot.admin_ids)
    await dp.start_polling(bot)


if __name__ == '__main__':
    try:
        asyncio.run(main())
    except (KeyboardInterrupt, SystemExit):
        logger.error("Бот був вимкнений!")