bnovo plugin

scheduller

bot/management/commands/run_bot.py

@@ -5,63 +5,56 @@ from apscheduler.schedulers.asyncio import AsyncIOScheduler
 from django.core.management.base import BaseCommand
 from telegram.ext import Application
 from bot.utils.bot_setup import setup_bot
-from bot.utils.scheduler import setup_scheduler
-from dotenv import load_dotenv
-from bot.operations.users import show_users
-
-# Загрузка переменных окружения
-load_dotenv()
+from scheduler.tasks import load_tasks_to_scheduler
 
 
 class Command(BaseCommand):
-    help = "Запуск Telegram бота"
+    help = "Запуск Telegram бота и планировщика"
 
     def handle(self, *args, **options):
-        print("Запуск Telegram бота...")
-
-        # Настройка Django окружения
+        # Установка Django окружения
         os.environ.setdefault("DJANGO_SETTINGS_MODULE", "touchh.settings")
         django.setup()
 
-        # Создание приложения Telegram
+        # Создаем новый цикл событий
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+
+        # Настройка планировщика
+        scheduler = AsyncIOScheduler(event_loop=loop)
+        scheduler.start()
+
+        # Загрузка задач в планировщик
+        try:
+            load_tasks_to_scheduler(scheduler)
+        except Exception as e:
+            self.stderr.write(f"Ошибка при загрузке задач в планировщик: {e}")
+            return
+
+        # Настройка Telegram бота
         bot_token = os.getenv("TELEGRAM_BOT_TOKEN")
         if not bot_token:
             raise ValueError("Токен бота не найден в переменных окружения.")
         application = Application.builder().token(bot_token).build()
-
-        # Настройка бота и обработчиков
         setup_bot(application)
 
+        # Основная асинхронная функция
         async def main():
-            print("Настройка планировщика...")
-            scheduler = setup_scheduler()
-            scheduler.start()
-
+            await application.initialize()
+            await application.start()
+            await application.updater.start_polling()
+            self.stdout.write(self.style.SUCCESS("Telegram бот и планировщик успешно запущены."))
             try:
-                print("Инициализация Telegram бота...")
-                await application.initialize()  # Инициализация приложения
-                print("Бот запущен. Ожидание сообщений...")
-                await application.start()       # Запуск приложения
-                await application.updater.start_polling()  # Запуск обработки сообщений
-
-                # Бесконечный цикл для удержания приложения активным
                 while True:
-                    await asyncio.sleep(3600)  # Ожидание 1 час
-            except Exception as e:
-                print(f"Ошибка во время работы бота: {e}")
-            finally:
-                print("Остановка Telegram бота...")
-                await application.stop()  # Завершаем приложение перед shutdown
-                print("Остановка планировщика...")
-                scheduler.shutdown(wait=False)
-                print("Планировщик остановлен.")
+                    await asyncio.sleep(3600)
+            except asyncio.CancelledError:
+                await application.stop()
+                scheduler.shutdown()
 
+        # Запуск асинхронной программы
         try:
-            asyncio.run(main())
-        except RuntimeError as e:
-            if str(e) == "This event loop is already running":
-                print("Цикл событий уже запущен. Используем другой подход для запуска.")
-                loop = asyncio.get_event_loop()
-                loop.run_until_complete(main())
-            else:
-                raise
+            loop.run_until_complete(main())
+        except KeyboardInterrupt:
+            self.stdout.write(self.style.ERROR("Завершение работы Telegram бота и планировщика"))
+        finally:
+            loop.close()