smart-bot-factory 1.1.1__py3-none-any.whl

smart_bot_factory/admin/admin_manager.py

@@ -0,0 +1,156 @@
+import logging
+from typing import Dict, List, Set
+
+from aiogram.types import User
+
+logger = logging.getLogger(__name__)
+
+
+class AdminManager:
+    """Управление администраторами бота"""
+
+    def __init__(self, config, supabase_client):
+        self.config = config
+        self.supabase = supabase_client
+        self.admin_ids: Set[int] = set(config.ADMIN_TELEGRAM_IDS)
+        self.admin_modes: Dict[int, bool] = {}  # admin_id -> is_in_admin_mode
+
+        logger.info(f"Инициализирован менеджер админов: {len(self.admin_ids)} админов")
+
+    async def sync_admins_from_config(self):
+        """Синхронизирует админов из конфига с базой данных"""
+        if not self.admin_ids:
+            logger.warning("Нет админов в конфигурации")
+            return
+
+        try:
+            for admin_id in self.admin_ids:
+                await self.supabase.sync_admin(
+                    {
+                        "telegram_id": admin_id,
+                        "username": None,  # будет обновлено при первом сообщении
+                        "first_name": None,
+                        "last_name": None,
+                    }
+                )
+
+                # Устанавливаем режим администратора по умолчанию
+                if admin_id not in self.admin_modes:
+                    self.admin_modes[admin_id] = True
+
+            logger.info(f"Синхронизированы админы: {self.admin_ids}")
+
+        except Exception as e:
+            logger.error(f"Ошибка синхронизации админов: {e}")
+            raise
+
+    async def update_admin_info(self, user: User):
+        """Обновляет информацию об админе"""
+        if not self.is_admin(user.id):
+            return
+
+        try:
+            await self.supabase.sync_admin(
+                {
+                    "telegram_id": user.id,
+                    "username": user.username,
+                    "first_name": user.first_name,
+                    "last_name": user.last_name,
+                }
+            )
+
+        except Exception as e:
+            logger.error(f"Ошибка обновления информации админа {user.id}: {e}")
+
+    def is_admin(self, telegram_id: int) -> bool:
+        """Проверяет, является ли пользователь админом"""
+        return telegram_id in self.admin_ids
+
+    def is_in_admin_mode(self, telegram_id: int) -> bool:
+        """Проверяет, находится ли админ в режиме администратора"""
+        if not self.is_admin(telegram_id):
+            return False
+        return self.admin_modes.get(telegram_id, True)
+
+    def toggle_admin_mode(self, telegram_id: int) -> bool:
+        """Переключает режим админа. Возвращает новое состояние"""
+        if not self.is_admin(telegram_id):
+            return False
+
+        current_mode = self.admin_modes.get(telegram_id, True)
+        new_mode = not current_mode
+        self.admin_modes[telegram_id] = new_mode
+
+        logger.info(
+            f"Админ {telegram_id} переключен в режим: {'администратор' if new_mode else 'пользователь'}"
+        )
+        return new_mode
+
+    def set_admin_mode(self, telegram_id: int, is_admin_mode: bool):
+        """Устанавливает режим админа"""
+        if not self.is_admin(telegram_id):
+            return
+
+        self.admin_modes[telegram_id] = is_admin_mode
+        logger.info(
+            f"Режим админа {telegram_id} установлен: {'администратор' if is_admin_mode else 'пользователь'}"
+        )
+
+    async def get_active_admins(self) -> List[int]:
+        """Возвращает список активных админов в режиме администратора"""
+        return [
+            admin_id for admin_id in self.admin_ids if self.is_in_admin_mode(admin_id)
+        ]
+
+    def get_admin_mode_text(self, telegram_id: int) -> str:
+        """Возвращает текстовое описание режима админа"""
+        if not self.is_admin(telegram_id):
+            return "Не администратор"
+
+        if self.is_in_admin_mode(telegram_id):
+            return "👑 Режим администратора"
+        else:
+            return "👤 Режим пользователя"
+
+    def format_admin_status(self, telegram_id: int) -> str:
+        """Форматирует статус админа для отображения"""
+        if not self.is_admin(telegram_id):
+            return ""
+
+        mode = "👑 АДМИН" if self.is_in_admin_mode(telegram_id) else "👤 ПОЛЬЗ"
+        return f"[{mode}]"
+
+    async def notify_admins(self, message: str, exclude_admin: int = None):
+        """Отправляет уведомление всем активным админам"""
+        from ..handlers.handlers import get_global_var
+        bot = get_global_var("bot")
+
+        active_admins = await self.get_active_admins()
+
+        if exclude_admin:
+            active_admins = [aid for aid in active_admins if aid != exclude_admin]
+
+        sent_count = 0
+        for admin_id in active_admins:
+            try:
+                await bot.send_message(admin_id, message, parse_mode="Markdown")
+                sent_count += 1
+            except Exception as e:
+                logger.error(f"Ошибка отправки уведомления админу {admin_id}: {e}")
+
+        logger.info(f"Уведомление отправлено {sent_count} админам")
+        return sent_count
+
+    def get_stats(self) -> Dict[str, any]:
+        """Возвращает статистику по админам"""
+        total_admins = len(self.admin_ids)
+        active_admins = len(
+            [aid for aid in self.admin_ids if self.is_in_admin_mode(aid)]
+        )
+
+        return {
+            "total_admins": total_admins,
+            "active_admins": active_admins,
+            "admin_ids": list(self.admin_ids),
+            "modes": dict(self.admin_modes),
+        }

smart_bot_factory/admin/admin_tester.py

@@ -0,0 +1,157 @@
+"""
+Утилита для тестирования системы администрирования бота
+"""
+
+import asyncio
+import logging
+import sys
+
+from ..analytics.analytics_manager import AnalyticsManager
+from ..config import Config
+from ..core.conversation_manager import ConversationManager
+from ..integrations.supabase_client import SupabaseClient
+from .admin_manager import AdminManager
+from .timeout_checker import setup_bot_environment
+
+logger = logging.getLogger(__name__)
+
+
+async def test_admin_system(bot_name: str = "growthmed-october-24") -> bool:
+    """
+    Тестирует систему администрирования бота
+
+    Args:
+        bot_name: Имя бота для тестирования
+
+    Returns:
+        bool: True если все тесты пройдены, False если найдены проблемы
+    """
+    logger.info(f"🚀 Тестирование системы администрирования: {bot_name}")
+    logger.info(f"🤖 Bot ID будет автоопределен как: {bot_name}\n")
+
+    # Настраиваем окружение для бота (автоматически устанавливает BOT_ID)
+    config_dir = setup_bot_environment(bot_name)
+    if not config_dir:
+        return False
+
+    # Инициализируем конфигурацию
+    config = Config()
+    logger.info("📋 Конфигурация:")
+    logger.info(f"   BOT_ID: {config.BOT_ID}")
+    logger.info(
+        f"   ADMIN_SESSION_TIMEOUT_MINUTES: {config.ADMIN_SESSION_TIMEOUT_MINUTES}"
+    )
+    logger.info(f"   PROMT_FILES_DIR: {config.PROMT_FILES_DIR}")
+    logger.info(f"   Найдено промпт-файлов: {len(config.PROMPT_FILES)}")
+    logger.info("")
+
+    # Проверяем админов
+    if not config.ADMIN_TELEGRAM_IDS:
+        logger.warning("⚠️ Админы не настроены (ADMIN_TELEGRAM_IDS пуст)")
+        return False
+
+    logger.info(f"👑 Админы: {config.ADMIN_TELEGRAM_IDS}")
+    logger.info("")
+
+    # Проверяем подключение к БД
+    try:
+        supabase_client = SupabaseClient(config.SUPABASE_URL, config.SUPABASE_KEY)
+        await supabase_client.initialize()
+        logger.info("✅ Подключение к Supabase успешно")
+
+        # Проверяем таблицы
+        tables = [
+            "sales_admins",
+            "admin_user_conversations",
+            "session_events",
+            "sales_chat_sessions",
+            "sales_messages",
+        ]
+
+        logger.info("\n📊 Проверка таблиц:")
+        for table in tables:
+            try:
+                supabase_client.client.table(table).select("*").limit(1).execute()
+                logger.info(f"   ✅ {table}")
+            except Exception as e:
+                logger.error(f"   ❌ {table}: {e}")
+                return False
+
+        # Проверяем AdminManager
+        admin_manager = AdminManager(config, supabase_client)
+        logger.info(
+            f"\n👑 AdminManager инициализирован ({len(admin_manager.admin_ids)} админов)"
+        )
+
+        # Проверяем ConversationManager
+        conversation_manager = ConversationManager(supabase_client, admin_manager)
+        logger.info("✅ ConversationManager инициализирован")
+
+        # Проверяем AnalyticsManager
+        analytics_manager = AnalyticsManager(supabase_client)
+
+        # Тестируем получение статистики
+        await analytics_manager.get_funnel_stats(1)
+        logger.info("✅ AnalyticsManager работает")
+
+        # Проверяем активные диалоги
+        conversations = await conversation_manager.get_active_conversations()
+        logger.info(f"\n💬 Активные диалоги: {len(conversations)}")
+
+        if conversations:
+            for conv in conversations:
+                logger.info(
+                    f"   • Диалог {conv['id']}: админ {conv['admin_id']} с пользователем {conv['user_id']}"
+                )
+        else:
+            logger.info("   Нет активных диалогов")
+            logger.info("   💡 Создайте диалог командой /чат USER_ID для тестирования")
+
+        # Проверяем форматирование диалогов
+        if conversations:
+            formatted = conversation_manager.format_active_conversations(conversations)
+            logger.info("\n📝 Форматирование диалогов:")
+            logger.info(formatted)
+
+        logger.info("\n✅ Админская система готова к работе")
+        return True
+
+    except Exception as e:
+        logger.error(f"❌ Ошибка тестирования: {e}")
+        logger.exception("Стек ошибки:")
+        return False
+
+
+def main():
+    """Точка входа для запуска из командной строки"""
+    # Настройка логирования
+    logging.basicConfig(level=logging.INFO, format="%(message)s")
+
+    logger.info("🔍 Утилита тестирования админской системы")
+    logger.info("Использование:")
+    logger.info("  python -m smart_bot_factory.admin_tester [bot_name]")
+    logger.info("  python -m smart_bot_factory.admin_tester growthmed-october-24")
+    logger.info("")
+
+    if len(sys.argv) > 1 and sys.argv[1] in ["-h", "--help", "help"]:
+        return
+
+    # Определяем какого бота тестировать
+    bot_name = "growthmed-october-24"  # по умолчанию
+    if len(sys.argv) > 1:
+        bot_name = sys.argv[1]
+
+    try:
+        success = asyncio.run(test_admin_system(bot_name))
+        if not success:
+            sys.exit(1)
+    except KeyboardInterrupt:
+        logger.info("\n⏹️ Прервано пользователем")
+    except Exception as e:
+        logger.error(f"\n💥 Критическая ошибка: {e}")
+        logger.exception("Стек ошибки:")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()