Initial public release

hydra/api/data.py

@@ -0,0 +1,50 @@
+from typing import Optional
+from fastapi import APIRouter, Depends, HTTPException, Query
+from hydra.api.auth import verify_api_key
+from hydra.data.storage.base import OhlcvStore
+
+router = APIRouter(prefix="/data")
+_store: Optional[OhlcvStore] = None
+
+
+def set_store(store: OhlcvStore) -> None:
+    global _store
+    _store = store
+
+
+@router.get("/candles")
+async def get_candles(
+    market: str,
+    symbol: str,
+    timeframe: str,
+    limit: int = Query(default=200, ge=1, le=1000),
+    since: Optional[int] = None,
+    _: str = Depends(verify_api_key),
+):
+    """Return OHLCV candles ordered by open_time ASC."""
+    if _store is None:
+        raise HTTPException(status_code=503, detail="Store not initialized")
+    candles = await _store.query(market, symbol, timeframe, limit=limit, since=since)
+    return [
+        {
+            "market": c.market,
+            "symbol": c.symbol,
+            "timeframe": c.timeframe,
+            "open_time": c.open_time,
+            "open": c.open,
+            "high": c.high,
+            "low": c.low,
+            "close": c.close,
+            "volume": c.volume,
+            "close_time": c.close_time,
+        }
+        for c in candles
+    ]
+
+
+@router.get("/symbols")
+async def get_symbols(_: str = Depends(verify_api_key)):
+    """Return distinct {market, symbol, timeframe} records being collected."""
+    if _store is None:
+        raise HTTPException(status_code=503, detail="Store not initialized")
+    return await _store.get_symbols()