89bdc6c552
- Move FRED credential registration to FastAPI lifespan (was fragile import-order-dependent side-effect) - Add noqa E402 annotations for imports after curl_cffi patch - Fix all return type hints: bare dict -> dict[str, Any] - Move yfinance import to module level (was inline in functions) - Fix datetime.now() -> datetime.now(tz=timezone.utc) in openbb_service - Add try/except error handling to Group B service functions - Fix dict mutation in relative_rotation (immutable pattern) - Extract _classify_rrg_quadrant helper function - Fix type builtin shadow in routes_economy (type -> gdp_type) - Fix falsy int guard (if year: -> if year is not None:) - Remove user input echo from error messages
98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
import logging
|
|
from contextlib import asynccontextmanager
|
|
|
|
import uvicorn
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
# Patch curl_cffi to use safari TLS fingerprint instead of chrome.
|
|
# curl_cffi's chrome impersonation triggers BoringSSL SSL_ERROR_SYSCALL on
|
|
# some networks; safari works reliably. This must happen before any import
|
|
# that creates a curl_cffi Session (yfinance, openbb).
|
|
import curl_cffi.requests as _cffi_requests
|
|
|
|
_orig_session_init = _cffi_requests.Session.__init__
|
|
|
|
|
|
def _patched_session_init(self, *args, **kwargs):
|
|
if kwargs.get("impersonate") == "chrome":
|
|
kwargs["impersonate"] = "safari"
|
|
_orig_session_init(self, *args, **kwargs)
|
|
|
|
|
|
_cffi_requests.Session.__init__ = _patched_session_init
|
|
|
|
from openbb import obb # noqa: E402 - must be after curl_cffi patch
|
|
|
|
from config import settings # noqa: E402
|
|
from routes import router # noqa: E402
|
|
from routes_calendar import router as calendar_router # noqa: E402
|
|
from routes_economy import router as economy_router # noqa: E402
|
|
from routes_fixed_income import router as fixed_income_router # noqa: E402
|
|
from routes_macro import router as macro_router # noqa: E402
|
|
from routes_market import router as market_router # noqa: E402
|
|
from routes_quantitative import router as quantitative_router # noqa: E402
|
|
from routes_regulators import router as regulators_router # noqa: E402
|
|
from routes_sentiment import router as sentiment_router # noqa: E402
|
|
from routes_shorts import router as shorts_router # noqa: E402
|
|
from routes_surveys import router as surveys_router # noqa: E402
|
|
from routes_technical import router as technical_router # noqa: E402
|
|
|
|
logging.basicConfig(
|
|
level=settings.log_level.upper(),
|
|
format="%(asctime)s %(levelname)s %(name)s: %(message)s",
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""Register provider credentials once at startup."""
|
|
if settings.fred_api_key:
|
|
obb.user.credentials.fred_api_key = settings.fred_api_key
|
|
logger.info("FRED API key registered")
|
|
yield
|
|
|
|
|
|
app = FastAPI(
|
|
title="OpenBB Investment Analysis API",
|
|
version="0.1.0",
|
|
description="REST API for stock data and rule-based investment analysis, powered by OpenBB SDK.",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=settings.cors_origins,
|
|
allow_credentials=False,
|
|
allow_methods=["GET", "POST"],
|
|
allow_headers=["Content-Type", "Authorization"],
|
|
)
|
|
|
|
app.include_router(router)
|
|
app.include_router(sentiment_router)
|
|
app.include_router(macro_router)
|
|
app.include_router(technical_router)
|
|
app.include_router(quantitative_router)
|
|
app.include_router(calendar_router)
|
|
app.include_router(market_router)
|
|
app.include_router(shorts_router)
|
|
app.include_router(fixed_income_router)
|
|
app.include_router(economy_router)
|
|
app.include_router(surveys_router)
|
|
app.include_router(regulators_router)
|
|
|
|
|
|
@app.get("/health", response_model=dict[str, str])
|
|
async def health() -> dict[str, str]:
|
|
return {"status": "ok"}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
uvicorn.run(
|
|
"main:app",
|
|
host=settings.host,
|
|
port=settings.port,
|
|
reload=settings.debug,
|
|
)
|