Spaces:
Running
Running
File size: 4,163 Bytes
09801ca | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 | """
DataVision — Application Factory
Creates and configures the FastAPI application.
Mounts both legacy v1 routes and new v2 routes for gradual migration.
"""
import logging
import sys
import os
from contextlib import asynccontextmanager
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
# Ensure the backend directory is on the path for legacy imports
BACKEND_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
if BACKEND_DIR not in sys.path:
sys.path.insert(0, BACKEND_DIR)
from app.core.config import get_settings
from app.database.session import dispose_engine
logger = logging.getLogger(__name__)
@asynccontextmanager
async def lifespan(app: FastAPI):
"""Application startup and shutdown lifecycle."""
logger.info("🚀 DataVision AI starting up...")
yield
logger.info("🛑 DataVision AI shutting down...")
await dispose_engine()
def create_app() -> FastAPI:
"""Create and configure the FastAPI application."""
settings = get_settings()
app = FastAPI(
title=settings.APP_NAME,
version=settings.APP_VERSION,
description="DataVision — Production-Grade AI Data Analytics Platform",
docs_url="/api/docs" if settings.is_development else None,
redoc_url="/api/redoc" if settings.is_development else None,
lifespan=lifespan,
)
# ── CORS ────────────────────────────────────────────────────
app.add_middleware(
CORSMiddleware,
allow_origins=settings.cors_origins_list,
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# ── New V2 API Routes ───────────────────────────────────────
from app.api.v2.router import router as v2_router
app.include_router(v2_router)
# ── Legacy V1 API Routes ────────────────────────────────────
# Mount the existing v1 application for backward compatibility
try:
from main import app as legacy_app
# Re-mount all legacy routes under the new app
for route in legacy_app.routes:
if hasattr(route, 'path') and route.path.startswith('/api/v1'):
app.routes.append(route)
logger.info("✅ Legacy v1 routes mounted")
except Exception as e:
logger.warning(f"⚠️ Could not mount legacy v1 routes: {e}")
# This is expected during initial development/testing
# The legacy app may have import issues until fully integrated
# ── Health Check ────────────────────────────────────────────
@app.get("/health", tags=["System"])
async def health_check():
return {
"status": "healthy",
"version": settings.APP_VERSION,
"environment": settings.ENVIRONMENT,
}
@app.get("/api/v2/health", tags=["System"])
async def v2_health():
return {
"status": "healthy",
"api_version": "v2",
"app_version": settings.APP_VERSION,
}
# ── Static Files ────────────────────────────────────────────
possible_static_dirs = [
os.path.join(BACKEND_DIR, "static"),
os.path.join(os.path.dirname(BACKEND_DIR), "static"),
os.path.join(os.path.dirname(BACKEND_DIR), "frontend", "dist"),
]
for static_dir in possible_static_dirs:
if os.path.isdir(static_dir) and os.path.exists(os.path.join(static_dir, "index.html")):
logger.info(f"✅ Serving static frontend from: {static_dir}")
app.mount("/", StaticFiles(directory=static_dir, html=True), name="static")
break
return app
# Module-level app instance for uvicorn
app = create_app()
|