aiform_dev/backend/app/main.py

"""
ERV Insurance Platform - FastAPI Backend
"""
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from contextlib import asynccontextmanager
import logging

from .config import settings
from .services.database import db
from .services.redis_service import redis_service
from .services.rabbitmq_service import rabbitmq_service
from .services.policy_service import policy_service
from .services.s3_service import s3_service
from .api import sms, claims, policy, upload, draft

# Настройка логирования
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)


@asynccontextmanager
async def lifespan(app: FastAPI):
    """
    Lifecycle events: startup and shutdown
    """
    # STARTUP
    logger.info("🚀 Starting ERV Platform...")
    
    try:
        # Подключаем PostgreSQL
        await db.connect()
    except Exception as e:
        logger.warning(f"⚠️ PostgreSQL not available: {e}")
    
    try:
        # Подключаем Redis
        await redis_service.connect()
    except Exception as e:
        logger.warning(f"⚠️ Redis not available: {e}")
    
    try:
        # Подключаем RabbitMQ
        await rabbitmq_service.connect()
    except Exception as e:
        logger.warning(f"⚠️ RabbitMQ not available: {e}")
    
    try:
        # Подключаем MySQL (для проверки полисов)
        await policy_service.connect()
    except Exception as e:
        logger.warning(f"⚠️ MySQL Policy DB not available: {e}")
    
    try:
        # Подключаем S3 (для загрузки файлов)
        s3_service.connect()
    except Exception as e:
        logger.warning(f"⚠️ S3 storage not available: {e}")
    
    logger.info("✅ ERV Platform started successfully!")
    
    yield
    
    # SHUTDOWN
    logger.info("🛑 Shutting down ERV Platform...")
    
    await db.disconnect()
    await redis_service.disconnect()
    await rabbitmq_service.disconnect()
    await policy_service.close()
    
    logger.info("👋 ERV Platform stopped")


# Создаём FastAPI приложение
app = FastAPI(
    title="ERV Insurance Platform API",
    description="API для обработки страховых обращений",
    version="1.0.0",
    lifespan=lifespan
)

# CORS
app.add_middleware(
    CORSMiddleware,
    allow_origins=settings.cors_origins_list,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# API Routes
app.include_router(sms.router)
app.include_router(claims.router)
app.include_router(policy.router)
app.include_router(upload.router)
app.include_router(draft.router)


@app.get("/")
async def root():
    """Главная страница API"""
    return {
        "message": "🚀 ERV Insurance Platform API",
        "version": "1.0.0",
        "status": "running",
        "docs": "http://147.45.146.17:8100/docs"
    }


@app.get("/health")
async def health():
    """Health check - проверка всех сервисов"""
    health_status = {
        "status": "ok",
        "message": "API работает!",
        "services": {}
    }
    
    # Проверка PostgreSQL
    try:
        pg_healthy = await db.health_check()
        health_status["services"]["postgresql"] = {
            "status": "✅ healthy" if pg_healthy else "❌ unhealthy",
            "connected": pg_healthy
        }
    except:
        health_status["services"]["postgresql"] = {
            "status": "❌ unavailable",
            "connected": False
        }
    
    # Проверка Redis
    try:
        redis_healthy = await redis_service.health_check()
        health_status["services"]["redis"] = {
            "status": "✅ healthy" if redis_healthy else "❌ unhealthy",
            "connected": redis_healthy
        }
    except:
        health_status["services"]["redis"] = {
            "status": "❌ unavailable",
            "connected": False
        }
    
    # Проверка RabbitMQ
    try:
        rabbitmq_healthy = await rabbitmq_service.health_check()
        health_status["services"]["rabbitmq"] = {
            "status": "✅ healthy" if rabbitmq_healthy else "❌ unhealthy",
            "connected": rabbitmq_healthy
        }
    except:
        health_status["services"]["rabbitmq"] = {
            "status": "❌ unavailable",
            "connected": False
        }
    
    # Общий статус
    all_healthy = all(
        service.get("connected", False) 
        for service in health_status["services"].values()
    )
    
    if not all_healthy:
        health_status["status"] = "degraded"
        health_status["message"] = "⚠️ Некоторые сервисы недоступны"
    
    return health_status


@app.get("/api/v1/test")
async def test():
    """Тестовый endpoint"""
    return {
        "success": True,
        "message": "✅ Backend API работает!",
        "services": {
            "redis": "localhost:6379",
            "postgres": "147.45.189.234:5432",
            "ocr": "147.45.146.17:8001",
            "rabbitmq": "185.197.75.249:5672"
        }
    }


@app.get("/api/v1/info")
async def info():
    """Информация о платформе"""
    return {
        "platform": "ERV Insurance Claims",
        "version": "1.0.0",
        "tech_stack": {
            "backend": "Python FastAPI",
            "frontend": "React TypeScript",
            "database": "PostgreSQL + MySQL",
            "cache": "Redis",
            "queue": "RabbitMQ",
            "storage": "S3 Timeweb"
        },
        "features": [
            "OCR документов (паспорт, билеты)",
            "AI автозаполнение (Gemini Vision)",
            "Проверка рейсов (FlightAware)",
            "СБП выплаты",
            "Интеграция с CRM Vtiger"
        ]
    }


if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8100)