main.py

# distudio_bot/main.py
import asyncio
import logging

from aiogram import Bot, Dispatcher
from aiogram.client.default import DefaultBotProperties
from aiogram.enums import ParseMode

from core.config import config
from core.logging_config import logger
from database.migrations import init_db

# === Роутеры ===
from handlers.start import router as start_router
from handlers.user_handlers import router as user_handlers_router
from handlers.admin_handlers import router as admin_handlers_router
from handlers.developer_handlers import router as developer_handlers_router

# === Сервисы ===
from services.notification_service import NotificationService
from services.backup_service import BackupService


async def start_scheduled_broadcast_loop(notification_service: NotificationService):
    """Фоновый цикл для отправки запланированных рассылок каждую минуту"""
    logger.info("⏰ Сервис запланированных рассылок запущен (проверка каждую минуту)")
    
    while True:
        try:
            await notification_service.send_scheduled_broadcasts()
            await asyncio.sleep(60)  # Проверяем каждую минуту
        except Exception as e:
            logger.error(f"Ошибка в цикле запланированных рассылок: {e}")
            await asyncio.sleep(60)


async def main():
    # Инициализация базы данных
    await init_db()
    logger.info("✅ База данных инициализирована")

    # Создание бота и диспетчера
    bot = Bot(
        token=config.BOT_TOKEN,
        default=DefaultBotProperties(parse_mode=ParseMode.HTML)
    )
    dp = Dispatcher()

    # Передаём общие данные в диспетчер
    dp["user_count"] = 0  # будет обновляться при рассылке

    # Подключение роутеров
    dp.include_router(start_router)
    dp.include_router(user_handlers_router)
    dp.include_router(admin_handlers_router)
    dp.include_router(developer_handlers_router)


    logger.info("🚀 Бот запускается...")

    # Запуск фоновых задач
    # Передаем только bot, сессию сервис получит сам через get_db_session()
    notification_service = NotificationService(bot, None)
    reminder_task = asyncio.create_task(notification_service.start_reminder_loop())
    scheduled_broadcast_task = asyncio.create_task(start_scheduled_broadcast_loop(notification_service))
    backup_task = asyncio.create_task(BackupService(bot).auto_backup_loop())

    try:
        # Запуск polling
        await dp.start_polling(bot, allowed_updates=dp.resolve_used_update_types())
    except Exception as e:
        logger.error(f"❌ Ошибка при запуске бота: {e}")
    finally:
        # Отмена фоновых задач
        reminder_task.cancel()
        scheduled_broadcast_task.cancel()
        backup_task.cancel()
        await bot.session.close()
        logger.info("🛑 Бот остановлен.")


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        logger.info("👋 Бот остановлен вручную.")