smart-bot-factory 0.2.10__py3-none-any.whl → 0.3.1__py3-none-any.whl

smart_bot_factory/core/router.py

@@ -57,7 +57,7 @@ class EventRouter:
             return wrapper
         return decorator
     
-    def schedule_task(self, task_name: str, notify: bool = False, smart_check: bool = True, once_only: bool = True, delay: Union[str, int] = None, event_type: str = None):
+    def schedule_task(self, task_name: str, notify: bool = False, smart_check: bool = True, once_only: bool = True, delay: Union[str, int] = None, event_type: Union[str, Callable] = None):
         """
         Декоратор для регистрации запланированной задачи в роутере
         
@@ -67,7 +67,9 @@ class EventRouter:
             smart_check: Использовать ли умную проверку
             once_only: Выполнять ли только один раз
             delay: Время задержки в удобном формате (например, "1h 30m", "45m", 3600) - ОБЯЗАТЕЛЬНО
-            event_type: Тип события для напоминания (например, 'appointment_booking') - ОПЦИОНАЛЬНО
+            event_type: Источник времени события - ОПЦИОНАЛЬНО:
+                - str: Тип события для поиска в БД (например, 'appointment_booking')
+                - Callable: Функция async def(user_id, user_data) -> datetime
         """
         def decorator(func: Callable) -> Callable:
             # Время ОБЯЗАТЕЛЬНО должно быть указано
@@ -115,7 +117,7 @@ class EventRouter:
             return wrapper
         return decorator
     
-    def global_handler(self, handler_type: str, notify: bool = False, once_only: bool = True, delay: Union[str, int] = None):
+    def global_handler(self, handler_type: str, notify: bool = False, once_only: bool = True, delay: Union[str, int] = None, event_type: Union[str, Callable] = None):
         """
         Декоратор для регистрации глобального обработчика в роутере
         
@@ -147,7 +149,8 @@ class EventRouter:
                 'notify': notify,
                 'once_only': once_only,
                 'router': self.name,
-                'default_delay': default_delay_seconds
+                'default_delay': default_delay_seconds,
+                'event_type': event_type  # Добавляем event_type для глобальных обработчиков
             }
             
             logger.info(f"🌍 Роутер {self.name}: зарегистрирован глобальный обработчик '{handler_type}': {func.__name__}")

smart_bot_factory/creation/bot_builder.py

@@ -317,25 +317,45 @@ class BotBuilder:
         Регистрирует Telegram роутер для обработки команд и сообщений
         
         Args:
-            telegram_router: TelegramRouter для регистрации
+            telegram_router: aiogram.Router для регистрации
+            
+        Example:
+            from aiogram import Router
+            from aiogram.filters import Command
+            
+            # Создаем обычный aiogram Router
+            my_router = Router(name="my_commands")
+            
+            @my_router.message(Command("price"))
+            async def price_handler(message: Message):
+                await message.answer("Наши цены...")
+            
+            # Регистрируем в боте
+            bot_builder.register_telegram_router(my_router)
         """
-        from ..core.telegram_router import TelegramRouter
+        from aiogram import Router as AiogramRouter
         
-        if not isinstance(telegram_router, TelegramRouter):
-            raise TypeError(f"Ожидается TelegramRouter, получен {type(telegram_router)}")
+        if not isinstance(telegram_router, AiogramRouter):
+            raise TypeError(f"Ожидается aiogram.Router, получен {type(telegram_router)}")
         
         self._telegram_routers.append(telegram_router)
-        logger.info(f"✅ Telegram роутер '{telegram_router.name}' зарегистрирован в боте {self.bot_id}")
+        router_name = getattr(telegram_router, 'name', 'unnamed')
+        logger.info(f"✅ Telegram роутер '{router_name}' зарегистрирован в боте {self.bot_id}")
     
     def register_telegram_routers(self, *telegram_routers):
         """
         Регистрирует несколько Telegram роутеров одновременно
         
         Args:
-            *telegram_routers: Произвольное количество TelegramRouter
+            *telegram_routers: Произвольное количество aiogram.Router
             
         Example:
-            bot_builder.register_telegram_routers(telegram_router1, telegram_router2)
+            from aiogram import Router
+            
+            router1 = Router(name="commands")
+            router2 = Router(name="callbacks")
+            
+            bot_builder.register_telegram_routers(router1, router2)
         """
         if not telegram_routers:
             logger.warning("⚠️ register_telegram_routers вызван без аргументов")
@@ -542,17 +562,23 @@ class BotBuilder:
         """
         Регистрирует фильтр отправки (может блокировать отправку пользователю)
         
-        Если фильтр возвращает False, сообщение НЕ отправляется
+        Если фильтр возвращает True, сообщение НЕ отправляется
         
         Args:
-            handler: async def(user_id: int, response_text: str) -> bool
+            handler: async def(user_id: int) -> bool
         
         Example:
             @bot_builder.filter_send
-            async def block_during_process(user_id, response_text):
+            async def block_during_process(user_id):
                 if is_processing(user_id):
-                    return False  # Не отправляем
-                return True  # Отправляем
+                    return True  # Блокируем отправку
+                return False  # Разрешаем отправку
+            
+            # Или совместимый с should_block_ai_response
+            @bot_builder.filter_send
+            async def should_block_ai_response(user_id):
+                # Ваша логика проверки
+                return user_is_blocked(user_id)  # True = блокировать
         """
         if not callable(handler):
             raise TypeError(f"Обработчик должен быть callable, получен {type(handler)}")
@@ -682,8 +708,9 @@ class BotBuilder:
             if self._telegram_routers:
                 logger.info(f"🔗 Подключаем {len(self._telegram_routers)} пользовательских Telegram роутеров")
                 for telegram_router in self._telegram_routers:
-                    dp.include_router(telegram_router.get_aiogram_router())
-                    logger.info(f"✅ Подключен Telegram роутер: {telegram_router.name}")
+                    dp.include_router(telegram_router)
+                    router_name = getattr(telegram_router, 'name', 'unnamed')
+                    logger.info(f"✅ Подключен Telegram роутер: {router_name}")
             
             # Настраиваем стандартные обработчики (меньший приоритет)
             setup_utils_handlers(dp)    # Утилитарные команды (/status, /help)
@@ -712,6 +739,12 @@ class BotBuilder:
             logger.info(f"   👑 Админов настроено: {len(self.config.ADMIN_TELEGRAM_IDS)}")
             logger.info(f"   📝 Загружено промптов: {len(self.config.PROMPT_FILES)}")
             
+            # Запускаем фоновый процессор событий
+            from ..core.decorators import background_event_processor
+            import asyncio
+            asyncio.create_task(background_event_processor())
+            logger.info("✅ Фоновый процессор событий запущен (проверка БД каждую минуту)")
+            
             # Четкое сообщение о запуске
             print(f"\n🤖 БОТ {self.bot_id.upper()} УСПЕШНО ЗАПУЩЕН!")
             print(f"📱 Telegram Bot ID: {self.config.BOT_ID}")

smart_bot_factory/handlers/handlers.py

@@ -64,9 +64,9 @@ async def start_handler(message: Message, state: FSMContext):
         logger.error(f"Ошибка при обработке /start: {e}")
         await send_message(message, "Произошла ошибка при инициализации. Попробуйте позже.")
 
-@router.message(Command(commands=["timeup"]))
+@router.message(Command(commands=["timeup", "вперед"]))
 async def timeup_handler(message: Message, state: FSMContext):
-    """Обработчик команды /timeup - тестирование запланированных событий"""
+    """Обработчик команды /timeup (или /вперед) - тестирование запланированных событий"""
     from ..core.decorators import process_scheduled_event, update_event_result
     from datetime import datetime
     
@@ -753,8 +753,9 @@ async def process_user_message(message: Message, state: FSMContext, session_id:
         for filter_func in send_filters:
             try:
                 should_send = await filter_func(message.from_user.id)
-                if not should_send:
-                    logger.info(f"⛔ Фильтр '{filter_func.__name__}' заблокировал отправку")
+                if should_send:
+                    # True = блокируем (для совместимости с should_block_ai_response)
+                    logger.info(f"⛔ Фильтр '{filter_func.__name__}' заблокировал отправку (вернул True)")
                     return  # Не отправляем
             except Exception as e:
                 logger.error(f"❌ Ошибка в фильтре отправки '{filter_func.__name__}': {e}")

smart_bot_factory/router/__init__.py

@@ -3,9 +3,8 @@ Router модули smart_bot_factory
 """
 
 from ..core.router import EventRouter
-from ..core.telegram_router import TelegramRouter
 
 __all__ = [
-    'EventRouter',  # Роутер для событий
-    'TelegramRouter',  # Роутер для Telegram
+    'EventRouter',  # Роутер для событий (бизнес-логика)
+    # Для Telegram используйте aiogram.Router напрямую
 ]