update

src/quizzi/application/__main__.py

@@ -0,0 +1,97 @@
+import asyncio
+import logging
+
+from aiogram import Bot, Dispatcher
+from aiogram.client.default import DefaultBotProperties
+from aiogram.enums import ParseMode
+from aiogram_dialog import setup_dialogs
+from apscheduler.schedulers.asyncio import AsyncIOScheduler
+from dishka import make_async_container
+from dishka.integrations.aiogram import setup_dishka
+
+from quizzi.application.bot.admin_dialogs.main_menu import admin_menu_dialog
+from quizzi.application.bot.admin_dialogs.users import admin_users_dialog
+from quizzi.application.bot.creator_dialogs.main_menu import creator_menu_dialog
+from quizzi.application.bot.creator_dialogs.users import creator_users_dialog
+from quizzi.application.bot.handlers import router
+from quizzi.application.bot.middlewares.reject_not_admin import RejectNotAdminMiddleware
+from quizzi.application.bot.middlewares.reject_not_creator import RejectNotCreatorMiddleware
+from quizzi.application.bot.shared_dialogs.broadcast import shared_broadcast_dialog
+from quizzi.application.bot.shared_dialogs.create_test import shared_create_test_dialog
+from quizzi.application.bot.shared_dialogs.groups import shared_groups_dialog
+from quizzi.application.bot.shared_dialogs.templates import shared_templates_dialog
+from quizzi.application.bot.shared_dialogs.tests import shared_tests_dialog
+from quizzi.application.bot.user_dialogs.deeplink import deeplink_dialog
+from quizzi.application.bot.user_dialogs.main_menu import user_menu_dialog
+from quizzi.application.bot.user_dialogs.registration import registration_dialog
+from quizzi.application.bot.user_dialogs.take_test import take_test_dialog
+from quizzi.infrastructure.database.repo.user import UserRepository
+from quizzi.infrastructure.di import DatabaseProvider, SchedulerProvider
+from quizzi.infrastructure.utils.bot_commands import setup_bot_commands
+from quizzi.infrastructure.utils.config import Config
+
+
+async def main() -> None:
+    logging.basicConfig(
+        level=logging.INFO,
+        format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+    )
+    
+    config = Config.from_toml("config.toml")
+    
+    bot = Bot(
+        token=config.bot.token,
+        default=DefaultBotProperties(parse_mode=ParseMode.HTML)
+    )
+    
+    dp = Dispatcher()
+    
+    dp.include_routers(
+        router,
+        user_menu_dialog,
+        take_test_dialog,
+        registration_dialog,
+        deeplink_dialog,
+        shared_tests_dialog,
+        shared_groups_dialog,
+        shared_broadcast_dialog,
+        shared_templates_dialog,
+        shared_create_test_dialog,
+        admin_menu_dialog,
+        admin_users_dialog,
+        creator_menu_dialog,
+        creator_users_dialog,
+    )
+    
+    router.message.middleware(RejectNotAdminMiddleware())
+    router.message.middleware(RejectNotCreatorMiddleware())
+    
+    container = make_async_container(
+        DatabaseProvider(),
+        SchedulerProvider(),
+        context={Bot: bot, Config: config}
+    )
+    setup_dialogs(dp)
+    setup_dishka(container, dp, auto_inject=True)
+    
+    async with container() as request_container:
+        user_repo = await request_container.get(UserRepository)
+        await setup_bot_commands(bot, config, user_repo)
+    
+    scheduler = await container.get(AsyncIOScheduler)
+    scheduler.start()
+    
+    await bot.delete_webhook(drop_pending_updates=True)
+    
+    logging.info("Бот запущен")
+    logging.info("Планировщик задач запущен")
+    
+    try:
+        await dp.start_polling(bot)
+    finally:
+        scheduler.shutdown()
+        await bot.session.close()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())