install.py

#!/usr/bin/env python3
"""
Скрипт автоматической установки Pi2 Reactor Bot
"""

import sys
import subprocess
import os
import platform
from colorama import init, Fore, Style

# Инициализация colorama
try:
    init()
    colors_available = True
except ImportError:
    colors_available = False

def colored_print(text, color=None):
    """Цветной вывод с fallback"""
    if colors_available and color:
        print(f"{color}{text}{Style.RESET_ALL}")
    else:
        print(text)

def print_banner():
    """Баннер установщика"""
    banner = """
╔══════════════════════════════════════════════╗
║       Pi2 Reactor Bot - Установщик           ║
║                                              ║
║     🤖 Автоматическая установка всех         ║
║        необходимых компонентов               ║
╚══════════════════════════════════════════════╝
"""
    colored_print(banner, Fore.CYAN)

def check_python_version():
    """Проверка версии Python"""
    colored_print("🐍 Проверка версии Python...", Fore.YELLOW)
    
    if sys.version_info < (3, 8):
        colored_print("❌ Требуется Python 3.8 или выше!", Fore.RED)
        colored_print(f"   Текущая версия: {sys.version}", Fore.RED)
        return False
    
    colored_print(f"✅ Python {sys.version.split()[0]} - OK", Fore.GREEN)
    return True

def check_pip():
    """Проверка наличия pip"""
    colored_print("📦 Проверка pip...", Fore.YELLOW)
    
    try:
        subprocess.check_output([sys.executable, '-m', 'pip', '--version'])
        colored_print("✅ pip установлен", Fore.GREEN)
        return True
    except subprocess.CalledProcessError:
        colored_print("❌ pip не найден!", Fore.RED)
        return False

def update_pip():
    """Обновление pip"""
    colored_print("🔄 Обновление pip...", Fore.YELLOW)
    
    try:
        subprocess.check_call([
            sys.executable, '-m', 'pip', 'install', '--upgrade', 'pip'
        ])
        colored_print("✅ pip обновлен", Fore.GREEN)
        return True
    except subprocess.CalledProcessError as e:
        colored_print(f"⚠️ Не удалось обновить pip: {e}", Fore.YELLOW)
        return False

def install_requirements():
    """Установка зависимостей"""
    colored_print("📥 Установка зависимостей...", Fore.YELLOW)
    
    requirements = [
        'selenium>=4.16.0',
        'webdriver-manager>=4.0.1', 
        'requests>=2.31.0',
        'colorama>=0.4.6',
        'pillow>=10.1.0',
        'opencv-python>=4.8.1.78',
        'numpy>=1.25.2',
        'undetected-chromedriver>=3.5.4',
        'scikit-learn>=1.3.0'  # для кластеризации цветов
    ]
    
    failed_packages = []
    
    for package in requirements:
        try:
            colored_print(f"   Installing {package}...", Fore.BLUE)
            subprocess.check_call([
                sys.executable, '-m', 'pip', 'install', package
            ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
            colored_print(f"   ✅ {package}", Fore.GREEN)
        except subprocess.CalledProcessError:
            colored_print(f"   ❌ {package}", Fore.RED)
            failed_packages.append(package)
    
    if failed_packages:
        colored_print(f"\n⚠️ Не удалось установить:", Fore.YELLOW)
        for package in failed_packages:
            colored_print(f"   - {package}", Fore.RED)
        return False
    
    colored_print("\n✅ Все зависимости установлены успешно!", Fore.GREEN)
    return True

def check_chrome():
    """Проверка наличия Chrome"""
    colored_print("🌐 Проверка Google Chrome...", Fore.YELLOW)
    
    system = platform.system()
    chrome_paths = {
        'Windows': [
            r'C:\Program Files\Google\Chrome\Application\chrome.exe',
            r'C:\Program Files (x86)\Google\Chrome\Application\chrome.exe'
        ],
        'Darwin': [  # macOS
            '/Applications/Google Chrome.app/Contents/MacOS/Google Chrome'
        ],
        'Linux': [
            '/usr/bin/google-chrome',
            '/usr/bin/google-chrome-stable',
            '/usr/bin/chromium-browser'
        ]
    }
    
    if system in chrome_paths:
        for path in chrome_paths[system]:
            if os.path.exists(path):
                colored_print("✅ Google Chrome найден", Fore.GREEN)
                return True
    
    colored_print("⚠️ Google Chrome не найден", Fore.YELLOW)
    colored_print("   Установите Chrome: https://www.google.com/chrome/", Fore.YELLOW)
    return False

def create_directories():
    """Создание необходимых директорий"""
    colored_print("📁 Создание директорий...", Fore.YELLOW)
    
    directories = ['logs', 'sessions', 'screenshots']
    
    for directory in directories:
        try:
            os.makedirs(directory, exist_ok=True)
            colored_print(f"   ✅ {directory}/", Fore.GREEN)
        except Exception as e:
            colored_print(f"   ❌ {directory}/: {e}", Fore.RED)
    
    return True

def test_installation():
    """Тест установки"""
    colored_print("🧪 Тестирование установки...", Fore.YELLOW)
    
    test_imports = [
        ('selenium', 'webdriver'),
        ('cv2', None),
        ('PIL', 'Image'),
        ('numpy', None),
        ('requests', None),
        ('colorama', None),
        ('undetected_chromedriver', None)
    ]
    
    failed_imports = []
    
    for module_name, submodule in test_imports:
        try:
            if submodule:
                module = __import__(module_name, fromlist=[submodule])
                getattr(module, submodule)
            else:
                __import__(module_name)
            colored_print(f"   ✅ {module_name}", Fore.GREEN)
        except ImportError as e:
            colored_print(f"   ❌ {module_name}: {e}", Fore.RED)
            failed_imports.append(module_name)
    
    if failed_imports:
        colored_print(f"\n❌ Ошибки импорта модулей:", Fore.RED)
        for module in failed_imports:
            colored_print(f"   - {module}", Fore.RED)
        return False
    
    colored_print("\n✅ Все модули импортируются успешно!", Fore.GREEN)
    return True

def create_quick_start_script():
    """Создание скрипта быстрого запуска"""
    colored_print("🚀 Создание скрипта быстрого запуска...", Fore.YELLOW)
    
    script_content = '''#!/usr/bin/env python3
"""Quick Start для Pi2 Reactor Bot"""

import sys
import os

def main():
    print("🤖 Pi2 Reactor Bot - Quick Start")
    print("=" * 40)
    
    email = input("📧 Введите ваш email: ")
    
    print("\\n🎮 Выберите режим:")
    print("1. Быстрый")
    print("2. Сбалансированный (рекомендуется)")
    print("3. Осторожный")
    
    while True:
        choice = input("Выбор (1-3): ")
        if choice in ['1', '2', '3']:
            break
        print("Введите 1, 2 или 3")
    
    modes = {
        '1': '--fast',
        '2': '',
        '3': '--slow'
    }
    
    cmd = f'python run_bot.py --email {email} {modes[choice]}'
    print(f"\\n🚀 Запуск: {cmd}")
    
    os.system(cmd)

if __name__ == "__main__":
    main()
'''
    
    try:
        with open('quick_start.py', 'w', encoding='utf-8') as f:
            f.write(script_content)
        colored_print("✅ quick_start.py создан", Fore.GREEN)
        return True
    except Exception as e:
        colored_print(f"❌ Ошибка создания quick_start.py: {e}", Fore.RED)
        return False

def main():
    """Главная функция установщика"""
    print_banner()
    
    # Проверки системы
    if not check_python_version():
        return 1
    
    if not check_pip():
        colored_print("Установите pip: https://pip.pypa.io/en/stable/installation/", Fore.RED)
        return 1
    
    # Обновление pip
    update_pip()
    
    # Проверка Chrome
    check_chrome()
    
    # Установка зависимостей
    if not install_requirements():
        colored_print("\\n❌ Установка не завершена из-за ошибок", Fore.RED)
        return 1
    
    # Создание директорий
    create_directories()
    
    # Тестирование
    if not test_installation():
        colored_print("\\n❌ Тестирование не прошло", Fore.RED)
        return 1
    
    # Создание дополнительных скриптов
    create_quick_start_script()
    
    # Финальное сообщение
    colored_print("\\n" + "="*50, Fore.CYAN)
    colored_print("🎉 УСТАНОВКА ЗАВЕРШЕНА УСПЕШНО!", Fore.GREEN)
    colored_print("="*50, Fore.CYAN)
    
    colored_print("\\n📋 Что дальше:", Fore.YELLOW)
    colored_print("1. Запустите: python run_bot.py", Fore.WHITE)
    colored_print("2. Или используйте: python quick_start.py", Fore.WHITE)
    colored_print("3. Для помощи: python run_bot.py --help", Fore.WHITE)
    
    colored_print("\\n📚 Документация: README.md", Fore.BLUE)
    colored_print("🐛 Проблемы? Проверьте logs/", Fore.BLUE)
    
    colored_print("\\n🎮 Удачной игры! 🚀", Fore.MAGENTA)
    
    return 0

if __name__ == "__main__":
    try:
        sys.exit(main())
    except KeyboardInterrupt:
        colored_print("\\n🛑 Установка прервана пользователем", Fore.YELLOW)
        sys.exit(1)
    except Exception as e:
        colored_print(f"\\n💥 Критическая ошибка: {e}", Fore.RED)
        sys.exit(1)