start_system.py

#!/usr/bin/env python
"""
Скрипт для запуска всей системы управления задачами
"""
import subprocess
import sys
import os
import time
import signal
import threading
from pathlib import Path

class SystemManager:
    def __init__(self):
        self.processes = []
        self.running = True
        
    def start_process(self, name, command, cwd=None):
        """Запускает процесс"""
        print(f"🚀 Запуск {name}...")
        try:
            if isinstance(command, str):
                process = subprocess.Popen(command, shell=True, cwd=cwd)
            else:
                process = subprocess.Popen(command, cwd=cwd)
            
            self.processes.append((name, process))
            print(f"✅ {name} запущен (PID: {process.pid})")
            return process
        except Exception as e:
            print(f"❌ Ошибка запуска {name}: {e}")
            return None
    
    def check_redis(self):
        """Проверяет доступность Redis"""
        try:
            import redis
            r = redis.Redis(host='localhost', port=6379, db=0)
            r.ping()
            print("✅ Redis доступен")
            return True
        except Exception as e:
            print(f"❌ Redis недоступен: {e}")
            return False
    
    def start_redis(self):
        """Запускает Redis"""
        redis_path = Path("C:/Redis/redis-server.exe")
        if redis_path.exists():
            return self.start_process("Redis", str(redis_path))
        else:
            print("⚠️ Redis не найден в C:/Redis/")
            print("Скачайте Redis с https://github.com/microsoftarchive/redis/releases")
            return None
    
    def start_django(self):
        """Запускает Django сервер"""
        return self.start_process("Django", [sys.executable, "manage.py", "runserver"])
    
    def start_celery(self):
        """Запускает Celery Worker"""
        return self.start_process("Celery", [sys.executable, "run_celery.py"])
    
    def start_bot(self):
        """Запускает Telegram бота"""
        return self.start_process("Telegram Bot", [sys.executable, "run_bot.py"])
    
    def stop_all(self):
        """Останавливает все процессы"""
        print("\n🛑 Остановка всех процессов...")
        self.running = False
        
        for name, process in self.processes:
            try:
                print(f"Остановка {name}...")
                process.terminate()
                process.wait(timeout=5)
                print(f"✅ {name} остановлен")
            except subprocess.TimeoutExpired:
                print(f"⚠️ Принудительная остановка {name}...")
                process.kill()
            except Exception as e:
                print(f"❌ Ошибка остановки {name}: {e}")
        
        self.processes.clear()
    
    def monitor_processes(self):
        """Мониторит состояние процессов"""
        while self.running:
            time.sleep(5)
            for name, process in self.processes[:]:
                if process.poll() is not None:
                    print(f"⚠️ {name} завершился неожиданно")
                    self.processes.remove((name, process))
    
    def start_system(self):
        """Запускает всю систему"""
        print("🎯 Запуск системы управления задачами")
        print("=" * 50)
        
        # Проверяем Redis
        if not self.check_redis():
            print("Запуск Redis...")
            if not self.start_redis():
                print("❌ Не удалось запустить Redis")
                return False
            time.sleep(2)
        
        # Запускаем Django
        if not self.start_django():
            print("❌ Не удалось запустить Django")
            return False
        
        time.sleep(3)
        
        # Запускаем Celery
        if not self.start_celery():
            print("❌ Не удалось запустить Celery")
            return False
        
        time.sleep(2)
        
        # Запускаем Telegram бота
        if not self.start_bot():
            print("❌ Не удалось запустить Telegram бота")
            return False
        
        print("\n🎉 Система запущена!")
        print("📱 Веб-приложение: http://localhost:8000")
        print("🤖 Telegram бот: готов к работе")
        print("📊 Celery: обрабатывает уведомления")
        print("\nНажмите Ctrl+C для остановки")
        
        # Запускаем мониторинг
        monitor_thread = threading.Thread(target=self.monitor_processes)
        monitor_thread.daemon = True
        monitor_thread.start()
        
        try:
            while self.running:
                time.sleep(1)
        except KeyboardInterrupt:
            print("\nПолучен сигнал остановки...")
        finally:
            self.stop_all()
        
        return True


def main():
    """Основная функция"""
    # Обработчик сигналов для корректного завершения
    def signal_handler(signum, frame):
        print("\nПолучен сигнал завершения...")
        sys.exit(0)
    
    signal.signal(signal.SIGINT, signal_handler)
    signal.signal(signal.SIGTERM, signal_handler)
    
    manager = SystemManager()
    success = manager.start_system()
    
    if success:
        print("✅ Система успешно завершена")
    else:
        print("❌ Ошибка запуска системы")
        sys.exit(1)


if __name__ == "__main__":
    main()