feat: add 67 new endpoints across 10 feature groups
Prerequisite refactor: - Consolidate duplicate _to_dicts into shared obb_utils.to_list - Add fetch_historical and first_or_empty helpers to obb_utils Phase 1 - Local computation (no provider risk): - Group I: 12 technical indicators (ATR, ADX, Stoch, OBV, Ichimoku, Donchian, Aroon, CCI, Keltner, Fibonacci, A/D, Volatility Cones) - Group J: Sortino, Omega ratios + rolling stats (variance, stdev, mean, skew, kurtosis, quantile via generic endpoint) - Group H: ECB currency reference rates Phase 2 - FRED/Federal Reserve providers: - Group C: 10 fixed income endpoints (treasury rates, yield curve, auctions, TIPS, EFFR, SOFR, HQM, commercial paper, spot rates, spreads) - Group D: 11 economy endpoints (CPI, GDP, unemployment, PCE, money measures, CLI, HPI, FRED search, balance of payments, Fed holdings, FOMC documents) - Group E: 5 survey endpoints (Michigan, SLOOS, NFP, Empire State, BLS search) Phase 3 - SEC/stockgrid/FINRA providers: - Group B: 4 equity fundamental endpoints (management, dividends, SEC filings, company search) - Group A: 4 shorts/dark pool endpoints (short volume, FTD, short interest, OTC dark pool) - Group F: 3 index/ETF enhanced (S&P 500 multiples, index constituents, ETF N-PORT) Phase 4 - Regulators: - Group G: 5 regulatory endpoints (COT report, COT search, SEC litigation, institution search, CIK mapping) Security hardening: - Service-layer allowlists for all getattr dynamic dispatch - Regex validation on date, country, security_type, form_type params - Exception handling in fetch_historical - Callable guard on rolling stat dispatch Total: 32 existing + 67 new = 99 endpoints, all free providers.
This commit is contained in:
Yaojia Wang
@@ -6,28 +6,17 @@ from typing import Any
|
||||
|
||||
from openbb import obb
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
from obb_utils import fetch_historical, to_list
|
||||
|
||||
PROVIDER = "yfinance"
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
async def get_technical_indicators(
|
||||
symbol: str, days: int = 400
|
||||
) -> dict[str, Any]:
|
||||
"""Compute key technical indicators for a symbol."""
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
start = (datetime.now() - timedelta(days=days)).strftime("%Y-%m-%d")
|
||||
|
||||
# Fetch historical data first
|
||||
hist = await asyncio.to_thread(
|
||||
obb.equity.price.historical,
|
||||
symbol,
|
||||
start_date=start,
|
||||
provider=PROVIDER,
|
||||
)
|
||||
|
||||
if hist is None or hist.results is None:
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data available"}
|
||||
|
||||
result: dict[str, Any] = {"symbol": symbol}
|
||||
@@ -144,3 +133,266 @@ def _interpret_signals(data: dict[str, Any]) -> list[str]:
|
||||
signals.append("Death cross: SMA50 below SMA200 (bearish trend)")
|
||||
|
||||
return signals
|
||||
|
||||
|
||||
# --- Individual Indicator Functions (Phase 1, Group I) ---
|
||||
|
||||
|
||||
async def get_atr(symbol: str, length: int = 14, days: int = 400) -> dict[str, Any]:
|
||||
"""Average True Range -- volatility measurement for position sizing."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(
|
||||
obb.technical.atr, data=hist.results, length=length
|
||||
)
|
||||
latest = _extract_latest(result)
|
||||
return {
|
||||
"symbol": symbol,
|
||||
"length": length,
|
||||
"atr": latest.get(f"ATRr_{length}"),
|
||||
}
|
||||
except Exception:
|
||||
logger.warning("ATR failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute ATR"}
|
||||
|
||||
|
||||
async def get_adx(symbol: str, length: int = 14, days: int = 400) -> dict[str, Any]:
|
||||
"""Average Directional Index -- trend strength (>25 strong, <20 range-bound)."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(
|
||||
obb.technical.adx, data=hist.results, length=length
|
||||
)
|
||||
latest = _extract_latest(result)
|
||||
adx_val = latest.get(f"ADX_{length}")
|
||||
signal = "strong trend" if adx_val and adx_val > 25 else "range-bound"
|
||||
return {
|
||||
"symbol": symbol,
|
||||
"length": length,
|
||||
"adx": adx_val,
|
||||
"dmp": latest.get(f"DMP_{length}"),
|
||||
"dmn": latest.get(f"DMN_{length}"),
|
||||
"signal": signal,
|
||||
}
|
||||
except Exception:
|
||||
logger.warning("ADX failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute ADX"}
|
||||
|
||||
|
||||
async def get_stoch(
|
||||
symbol: str, fast_k: int = 14, slow_d: int = 3, slow_k: int = 3, days: int = 400,
|
||||
) -> dict[str, Any]:
|
||||
"""Stochastic Oscillator -- overbought/oversold momentum signal."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(
|
||||
obb.technical.stoch, data=hist.results,
|
||||
fast_k=fast_k, slow_d=slow_d, slow_k=slow_k,
|
||||
)
|
||||
latest = _extract_latest(result)
|
||||
k_val = latest.get(f"STOCHk_{fast_k}_{slow_d}_{slow_k}")
|
||||
d_val = latest.get(f"STOCHd_{fast_k}_{slow_d}_{slow_k}")
|
||||
signal = "neutral"
|
||||
if k_val is not None:
|
||||
if k_val > 80:
|
||||
signal = "overbought"
|
||||
elif k_val < 20:
|
||||
signal = "oversold"
|
||||
return {
|
||||
"symbol": symbol,
|
||||
"stoch_k": k_val,
|
||||
"stoch_d": d_val,
|
||||
"signal": signal,
|
||||
}
|
||||
except Exception:
|
||||
logger.warning("Stochastic failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute Stochastic"}
|
||||
|
||||
|
||||
async def get_obv(symbol: str, days: int = 400) -> dict[str, Any]:
|
||||
"""On-Balance Volume -- cumulative volume indicator for divergence detection."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(obb.technical.obv, data=hist.results)
|
||||
latest = _extract_latest(result)
|
||||
return {
|
||||
"symbol": symbol,
|
||||
"obv": latest.get("OBV"),
|
||||
}
|
||||
except Exception:
|
||||
logger.warning("OBV failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute OBV"}
|
||||
|
||||
|
||||
async def get_ichimoku(symbol: str, days: int = 400) -> dict[str, Any]:
|
||||
"""Ichimoku Cloud -- comprehensive trend system."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(obb.technical.ichimoku, data=hist.results)
|
||||
latest = _extract_latest(result)
|
||||
return {
|
||||
"symbol": symbol,
|
||||
"tenkan_sen": latest.get("ITS_9"),
|
||||
"kijun_sen": latest.get("IKS_26"),
|
||||
"senkou_span_a": latest.get("ISA_9"),
|
||||
"senkou_span_b": latest.get("ISB_26"),
|
||||
"chikou_span": latest.get("ICS_26"),
|
||||
}
|
||||
except Exception:
|
||||
logger.warning("Ichimoku failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute Ichimoku"}
|
||||
|
||||
|
||||
async def get_donchian(symbol: str, length: int = 20, days: int = 400) -> dict[str, Any]:
|
||||
"""Donchian Channels -- breakout detection system."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(
|
||||
obb.technical.donchian, data=hist.results, lower_length=length, upper_length=length,
|
||||
)
|
||||
latest = _extract_latest(result)
|
||||
return {
|
||||
"symbol": symbol,
|
||||
"length": length,
|
||||
"upper": latest.get(f"DCU_{length}_{length}"),
|
||||
"middle": latest.get(f"DCM_{length}_{length}"),
|
||||
"lower": latest.get(f"DCL_{length}_{length}"),
|
||||
}
|
||||
except Exception:
|
||||
logger.warning("Donchian failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute Donchian"}
|
||||
|
||||
|
||||
async def get_aroon(symbol: str, length: int = 25, days: int = 400) -> dict[str, Any]:
|
||||
"""Aroon Indicator -- trend direction and strength."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(
|
||||
obb.technical.aroon, data=hist.results, length=length,
|
||||
)
|
||||
latest = _extract_latest(result)
|
||||
up = latest.get(f"AROONU_{length}")
|
||||
down = latest.get(f"AROOND_{length}")
|
||||
osc = latest.get(f"AROONOSC_{length}")
|
||||
return {
|
||||
"symbol": symbol,
|
||||
"length": length,
|
||||
"aroon_up": up,
|
||||
"aroon_down": down,
|
||||
"aroon_oscillator": osc,
|
||||
}
|
||||
except Exception:
|
||||
logger.warning("Aroon failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute Aroon"}
|
||||
|
||||
|
||||
async def get_cci(symbol: str, length: int = 14, days: int = 400) -> dict[str, Any]:
|
||||
"""Commodity Channel Index -- cyclical trend identification."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(
|
||||
obb.technical.cci, data=hist.results, length=length,
|
||||
)
|
||||
latest = _extract_latest(result)
|
||||
cci_val = latest.get(f"CCI_{length}_{0.015}")
|
||||
signal = "neutral"
|
||||
if cci_val is not None:
|
||||
if cci_val > 100:
|
||||
signal = "overbought"
|
||||
elif cci_val < -100:
|
||||
signal = "oversold"
|
||||
return {
|
||||
"symbol": symbol,
|
||||
"length": length,
|
||||
"cci": cci_val,
|
||||
"signal": signal,
|
||||
}
|
||||
except Exception:
|
||||
logger.warning("CCI failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute CCI"}
|
||||
|
||||
|
||||
async def get_kc(symbol: str, length: int = 20, days: int = 400) -> dict[str, Any]:
|
||||
"""Keltner Channels -- ATR-based volatility bands."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(
|
||||
obb.technical.kc, data=hist.results, length=length,
|
||||
)
|
||||
latest = _extract_latest(result)
|
||||
return {
|
||||
"symbol": symbol,
|
||||
"length": length,
|
||||
"upper": latest.get(f"KCUe_{length}_2"),
|
||||
"middle": latest.get(f"KCBe_{length}_2"),
|
||||
"lower": latest.get(f"KCLe_{length}_2"),
|
||||
}
|
||||
except Exception:
|
||||
logger.warning("Keltner failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute Keltner Channels"}
|
||||
|
||||
|
||||
async def get_fib(symbol: str, days: int = 120) -> dict[str, Any]:
|
||||
"""Fibonacci Retracement levels from recent price range."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(obb.technical.fib, data=hist.results)
|
||||
latest = _extract_latest(result)
|
||||
return {"symbol": symbol, **latest}
|
||||
except Exception:
|
||||
logger.warning("Fibonacci failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute Fibonacci"}
|
||||
|
||||
|
||||
async def get_ad(symbol: str, days: int = 400) -> dict[str, Any]:
|
||||
"""Accumulation/Distribution Line -- volume-based trend indicator."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(obb.technical.ad, data=hist.results)
|
||||
latest = _extract_latest(result)
|
||||
return {
|
||||
"symbol": symbol,
|
||||
"ad": latest.get("AD"),
|
||||
"ad_obv": latest.get("AD_OBV"),
|
||||
}
|
||||
except Exception:
|
||||
logger.warning("A/D failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute A/D Line"}
|
||||
|
||||
|
||||
async def get_cones(symbol: str, days: int = 365) -> dict[str, Any]:
|
||||
"""Volatility Cones -- realized volatility quantiles for options analysis."""
|
||||
hist = await fetch_historical(symbol, days)
|
||||
if hist is None:
|
||||
return {"symbol": symbol, "error": "No historical data"}
|
||||
try:
|
||||
result = await asyncio.to_thread(
|
||||
obb.technical.cones, data=hist.results,
|
||||
)
|
||||
items = to_list(result)
|
||||
return {"symbol": symbol, "cones": items}
|
||||
except Exception:
|
||||
logger.warning("Volatility cones failed for %s", symbol, exc_info=True)
|
||||
return {"symbol": symbol, "error": "Failed to compute volatility cones"}
|
||||
|
||||
Reference in New Issue
Block a user