""" Factory for direct exchange clients. """ from __future__ import annotations from typing import Any, Dict from app.services.live_trading.base import BaseRestClient, LiveTradingError from app.services.live_trading.binance import BinanceFuturesClient from app.services.live_trading.binance_spot import BinanceSpotClient from app.services.live_trading.okx import OkxClient from app.services.live_trading.bitget import BitgetMixClient from app.services.live_trading.bitget_spot import BitgetSpotClient from app.services.live_trading.bybit import BybitClient from app.services.live_trading.coinbase_exchange import CoinbaseExchangeClient from app.services.live_trading.kraken import KrakenClient from app.services.live_trading.kraken_futures import KrakenFuturesClient from app.services.live_trading.kucoin import KucoinSpotClient, KucoinFuturesClient from app.services.live_trading.gate import GateSpotClient, GateUsdtFuturesClient from app.services.live_trading.bitfinex import BitfinexClient, BitfinexDerivativesClient def _get(cfg: Dict[str, Any], *keys: str) -> str: for k in keys: v = cfg.get(k) if v is None: continue s = str(v).strip() if s: return s return "" def create_client(exchange_config: Dict[str, Any], *, market_type: str = "swap") -> BaseRestClient: if not isinstance(exchange_config, dict): raise LiveTradingError("Invalid exchange_config") exchange_id = _get(exchange_config, "exchange_id", "exchangeId").lower() api_key = _get(exchange_config, "api_key", "apiKey") secret_key = _get(exchange_config, "secret_key", "secret") passphrase = _get(exchange_config, "passphrase", "password") mt = (market_type or exchange_config.get("market_type") or exchange_config.get("defaultType") or "swap").strip().lower() if mt in ("futures", "future", "perp", "perpetual"): mt = "swap" if exchange_id == "binance": if mt == "spot": base_url = _get(exchange_config, "base_url", "baseUrl") or "https://api.binance.com" return BinanceSpotClient(api_key=api_key, secret_key=secret_key, base_url=base_url) # Default to USDT-M futures base_url = _get(exchange_config, "base_url", "baseUrl") or "https://fapi.binance.com" return BinanceFuturesClient(api_key=api_key, secret_key=secret_key, base_url=base_url) if exchange_id == "okx": base_url = _get(exchange_config, "base_url", "baseUrl") or "https://www.okx.com" return OkxClient(api_key=api_key, secret_key=secret_key, passphrase=passphrase, base_url=base_url) if exchange_id == "bitget": base_url = _get(exchange_config, "base_url", "baseUrl") or "https://api.bitget.com" if mt == "spot": channel_api_code = _get(exchange_config, "channel_api_code", "channelApiCode") or "bntva" return BitgetSpotClient(api_key=api_key, secret_key=secret_key, passphrase=passphrase, base_url=base_url, channel_api_code=channel_api_code) return BitgetMixClient(api_key=api_key, secret_key=secret_key, passphrase=passphrase, base_url=base_url) if exchange_id == "bybit": base_url = _get(exchange_config, "base_url", "baseUrl") or "https://api.bybit.com" category = "spot" if mt == "spot" else "linear" recv_window_ms = int(exchange_config.get("recv_window_ms") or exchange_config.get("recvWindow") or 5000) return BybitClient(api_key=api_key, secret_key=secret_key, base_url=base_url, category=category, recv_window_ms=recv_window_ms) if exchange_id in ("coinbaseexchange", "coinbase_exchange"): base_url = _get(exchange_config, "base_url", "baseUrl") or "https://api.exchange.coinbase.com" if mt != "spot": raise LiveTradingError("CoinbaseExchange only supports spot market_type in this project") return CoinbaseExchangeClient(api_key=api_key, secret_key=secret_key, passphrase=passphrase, base_url=base_url) if exchange_id == "kraken": base_url = _get(exchange_config, "base_url", "baseUrl") or "https://api.kraken.com" if mt == "spot": return KrakenClient(api_key=api_key, secret_key=secret_key, base_url=base_url) # Futures/perp fut_url = _get(exchange_config, "futures_base_url", "futuresBaseUrl") or "https://futures.kraken.com" return KrakenFuturesClient(api_key=api_key, secret_key=secret_key, base_url=fut_url) if exchange_id == "kucoin": base_url = _get(exchange_config, "base_url", "baseUrl") or "https://api.kucoin.com" if mt == "spot": return KucoinSpotClient(api_key=api_key, secret_key=secret_key, passphrase=passphrase, base_url=base_url) fut_url = _get(exchange_config, "futures_base_url", "futuresBaseUrl") or "https://api-futures.kucoin.com" return KucoinFuturesClient(api_key=api_key, secret_key=secret_key, passphrase=passphrase, base_url=fut_url) if exchange_id == "gate": base_url = _get(exchange_config, "base_url", "baseUrl") or "https://api.gateio.ws" if mt == "spot": return GateSpotClient(api_key=api_key, secret_key=secret_key, base_url=base_url) # Default to USDT futures for swap return GateUsdtFuturesClient(api_key=api_key, secret_key=secret_key, base_url=base_url) if exchange_id == "bitfinex": base_url = _get(exchange_config, "base_url", "baseUrl") or "https://api.bitfinex.com" if mt == "spot": return BitfinexClient(api_key=api_key, secret_key=secret_key, base_url=base_url) return BitfinexDerivativesClient(api_key=api_key, secret_key=secret_key, base_url=base_url) raise LiveTradingError(f"Unsupported exchange_id: {exchange_id}")