app_shim.py

"""Top-level app shim for imports in tests and routes.

This module re-exports the real app from `backend.main` when available.
If that import fails, it falls back to a minimal FastAPI app that attempts
to include whichever routers can be imported.
"""
import os
from fastapi import FastAPI

# Try to import the real backend app. If successful, re-export it and set __file__
try:
    from backend import main as backend_main
    app = backend_main.app
    app_state = backend_main.app_state
    # Make this module appear to be backend/main.py for test assertions
    __file__ = os.path.join(os.path.dirname(__file__), "backend", "main.py")
except Exception:
    # Fallback shim (keeps previous behavior)
    app = FastAPI(title="Trading Backtester API (test shim)")

    # Minimal application state used by tests
    app_state = {
        "start_time": None,
        "models_loaded": {},
        "database_path": None,
        "active_websockets": set(),
        "backtests": {},
    }

    # Try to include routers from the `routes` package. If a specific router
    # fails to import, skip it so tests can still run.
    try:
        from backend.routes.health import router as health_router
        app.include_router(health_router)
    except Exception:
        pass

    try:
        from backend.routes.monitoring import router as monitoring_router
        app.include_router(monitoring_router)
    except Exception:
        pass

    try:
        from backend.routes.backtests import router as backtests_router
        app.include_router(backtests_router)
    except Exception:
        pass

    try:
        from backend.routes.models_endpoints import router as models_router
        app.include_router(models_router)
    except Exception:
        pass

    try:
        from backend.routes.portfolio import router as portfolio_router
        app.include_router(portfolio_router)
    except Exception:
        pass

    try:
        from backend.routes.data_endpoints import router as data_router
        app.include_router(data_router)
    except Exception:
        pass

    try:
        from backend.routes.scripts import router as scripts_router
        app.include_router(scripts_router)
    except Exception:
        pass

    try:
        from backend.routes.websocket import websocket_endpoint
        app.add_api_websocket_route("/ws", websocket_endpoint)
    except Exception:
        pass

__all__ = ["app", "app_state"]