103 lines
3.1 KiB
Python
103 lines
3.1 KiB
Python
"""FastAPI application — API Gateway for the trading bot."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from contextlib import asynccontextmanager
|
|
from typing import AsyncIterator
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from redis.asyncio import Redis
|
|
|
|
from services.api_gateway.auth.routes import router as auth_router
|
|
from services.api_gateway.config import ApiGatewayConfig
|
|
from shared.db import create_db
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def create_app(config: ApiGatewayConfig | None = None) -> FastAPI:
|
|
"""Build and configure the FastAPI application.
|
|
|
|
Parameters
|
|
----------
|
|
config:
|
|
Optional config override (useful for testing). If ``None``, a new
|
|
:class:`ApiGatewayConfig` is created from environment variables.
|
|
"""
|
|
if config is None:
|
|
config = ApiGatewayConfig()
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI) -> AsyncIterator[None]:
|
|
"""Start-up / shutdown hook — connect DB and Redis."""
|
|
# Database
|
|
engine, session_factory = create_db(config)
|
|
app.state.db_engine = engine
|
|
app.state.db_session_factory = session_factory
|
|
|
|
# Redis
|
|
app.state.redis = Redis.from_url(
|
|
config.redis_url, decode_responses=True
|
|
)
|
|
app.state.config = config
|
|
|
|
logger.info("API Gateway started")
|
|
yield
|
|
|
|
# Cleanup
|
|
await app.state.redis.aclose()
|
|
await engine.dispose()
|
|
logger.info("API Gateway stopped")
|
|
|
|
app = FastAPI(
|
|
title="Trading Bot API",
|
|
version="0.1.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
# CORS
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=config.cors_origins,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Auth routes (unauthenticated)
|
|
app.include_router(auth_router)
|
|
|
|
# Trading routes (authenticated) — imported lazily to avoid circular deps
|
|
from services.api_gateway.routes.portfolio import router as portfolio_router
|
|
from services.api_gateway.routes.trades import router as trades_router
|
|
from services.api_gateway.routes.signals import router as signals_router
|
|
from services.api_gateway.routes.strategies import router as strategies_router
|
|
from services.api_gateway.routes.news import router as news_router
|
|
from services.api_gateway.routes.controls import router as controls_router
|
|
from services.api_gateway.routes.backtest import router as backtest_router
|
|
|
|
app.include_router(portfolio_router)
|
|
app.include_router(trades_router)
|
|
app.include_router(signals_router)
|
|
app.include_router(strategies_router)
|
|
app.include_router(news_router)
|
|
app.include_router(controls_router)
|
|
app.include_router(backtest_router)
|
|
|
|
# WebSocket
|
|
from services.api_gateway.ws import router as ws_router
|
|
|
|
app.include_router(ws_router)
|
|
|
|
# Health check
|
|
@app.get("/health", tags=["health"])
|
|
async def health() -> dict:
|
|
return {"status": "ok"}
|
|
|
|
return app
|
|
|
|
|
|
# Convenience: allow ``uvicorn services.api_gateway.main:app``
|
|
app = create_app()
|