main.py

import asyncio
import logging

import structlog
from aiogram import Bot
from aiogram.exceptions import TelegramNetworkError
from aiogram.types import BotCommand

from app.bot import get_bot, get_dispatcher
from app.config import settings
from app.core.logging import configure_logging
from app.database.base import engine
from app.services.notifications import notify_admins_bot_started


async def set_commands(bot: Bot) -> None:
    logger = structlog.get_logger(__name__)
    commands = [
        BotCommand(command="start", description="Главное меню"),
        BotCommand(command="help", description="Помощь"),
        BotCommand(command="cancel", description="Отменить текущее действие"),
    ]
    for attempt in range(1, 4):
        try:
            await bot.set_my_commands(commands)
            return
        except TelegramNetworkError as exc:
            if attempt == 3:
                logger.warning(
                    "set_commands_skipped_network_error",
                    attempt=attempt,
                    error=str(exc),
                )
                return
            await asyncio.sleep(attempt * 2)


async def on_shutdown(bot: Bot) -> None:
    """Graceful shutdown hook called by Dispatcher on SIGTERM/SIGINT."""
    logger = structlog.get_logger(__name__)
    logger.info("bot_shutdown_started")
    try:
        await bot.session.close()
    except Exception as exc:
        logger.warning("bot_session_close_failed", error=str(exc))
    try:
        await engine.dispose()
    except Exception as exc:
        logger.warning("engine_dispose_failed", error=str(exc))
    logger.info("bot_shutdown_complete")


async def main() -> None:
    configure_logging(log_level=settings.LOG_LEVEL)
    logger = structlog.get_logger(__name__)
    bot = get_bot()
    dp = get_dispatcher()

    dp.shutdown.register(on_shutdown)
    try:
        await set_commands(bot)
        await notify_admins_bot_started(bot, settings.ADMIN_IDS, startup_source="main.py/polling")
        logger.info("bot_starting", admin_ids=len(settings.ADMIN_IDS))
        await dp.start_polling(bot, allowed_updates=dp.resolve_used_update_types())
    except Exception as exc:
        logger.exception("bot_polling_crashed", error=str(exc))
        raise


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        logging.getLogger(__name__).info("bot_stopped")