smart-bot-factory 0.3.6__py3-none-any.whl → 0.3.8__py3-none-any.whl

smart_bot_factory/admin/admin_manager.py

@@ -1,141 +1,156 @@
 import logging
-from typing import List, Dict, Set
+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
-                })
-                
+                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
-            })
-            
+            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 'пользователь'}")
+
+        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 'пользователь'}")
-    
+        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)]
-    
+        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 main import bot  # Импорт здесь чтобы избежать циклических импортов
-        
+        from main import \
+            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')
+                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)])
-        
+        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)
-        }
+            "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

@@ -4,138 +4,143 @@
 
 import asyncio
 import logging
-import json
-import re
 import sys
-import os
 
+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 ..core.conversation_manager import ConversationManager
-from ..analytics.analytics_manager import AnalyticsManager
 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(f"📋 Конфигурация:")
+    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"   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'
+            "sales_admins",
+            "admin_user_conversations",
+            "session_events",
+            "sales_chat_sessions",
+            "sales_messages",
         ]
-        
+
         logger.info("\n📊 Проверка таблиц:")
         for table in tables:
             try:
-                response = supabase_client.client.table(table).select('*').limit(1).execute()
+                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)} админов)")
-        
+        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)
-        
+
         # Тестируем получение статистики
-        funnel_stats = await analytics_manager.get_funnel_stats(1)
+        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']}")
+                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')
-    
+    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']:
+
+    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:
@@ -147,5 +152,6 @@ def main():
         logger.exception("Стек ошибки:")
         sys.exit(1)
 
+
 if __name__ == "__main__":
     main()