aria-code 0.52.0__py3-none-any.whl

aria_code/brokers/intl/ibkr_broker.py

@@ -0,0 +1,215 @@
+"""
+brokers/intl/ibkr_broker.py — Interactive Brokers 适配器
+==========================================================
+通过 ib_insync 连接 TWS 或 IB Gateway。
+
+安装：pip install ib_insync
+文档：https://ib-insync.readthedocs.io/
+
+配置示例::
+
+    {
+      "id":        "ibkr_us",
+      "type":      "ibkr",
+      "label":     "盈透美股",
+      "host":      "127.0.0.1",
+      "port":      7496,
+      "client_id": 1,
+      "readonly":  false
+    }
+
+端口说明：
+    TWS 实盘:  7496
+    TWS 模拟:  7497
+    Gateway 实盘: 4001
+    Gateway 模拟: 4002
+"""
+
+from __future__ import annotations
+
+from typing import Any, Dict, List
+
+from ..base import BrokerBase, AccountInfo, Position, Order, OrderResult
+
+
+class IBKRBroker(BrokerBase):
+    broker_type = "ibkr"
+    broker_name = "Interactive Brokers"
+    market      = "US"
+
+    def __init__(self, broker_id: str, config: Dict[str, Any]):
+        super().__init__(broker_id, config)
+        self._host      = config.get("host",      "127.0.0.1")
+        self._port      = int(config.get("port",  7496))
+        self._client_id = int(config.get("client_id", 1))
+        self._readonly  = bool(config.get("readonly", False))
+        self._ib        = None
+
+    def connect(self) -> bool:
+        try:
+            from ib_insync import IB
+        except ImportError:
+            raise ImportError("ib_insync 未安装。请运行: pip install ib_insync  并启动 TWS / IB Gateway")
+
+        try:
+            ib = IB()
+            ib.connect(self._host, self._port, clientId=self._client_id, readonly=self._readonly)
+            self._ib        = ib
+            self._connected = True
+            return True
+        except Exception as e:
+            raise RuntimeError(f"IBKR 连接失败: {e}") from e
+
+    def disconnect(self) -> None:
+        if self._ib:
+            try:
+                self._ib.disconnect()
+            except Exception:
+                pass
+        self._connected = False
+
+    def account_info(self) -> AccountInfo:
+        self._require_connected()
+        try:
+            summary = {v.tag: v.value for v in self._ib.accountSummary()}
+            account_id = summary.get("AccountCode", self._ib.managedAccounts()[0] if self._ib.managedAccounts() else "")
+            currency = summary.get("BaseCurrency", "USD")
+            netliq   = float(summary.get("NetLiquidation", 0))
+            cash     = float(summary.get("TotalCashValue",  0))
+            mv       = float(summary.get("GrossPositionValue", 0))
+            pnl      = float(summary.get("UnrealizedPnL",   0))
+        except Exception as e:
+            raise RuntimeError(f"IBKR 账户查询失败: {e}") from e
+
+        return AccountInfo(
+            broker_id=self.broker_id,
+            broker_type=self.broker_type,
+            label=self.label,
+            account_id=account_id,
+            currency=currency,
+            total_assets=netliq,
+            cash=cash,
+            market_value=mv,
+            pnl_total=pnl,
+        )
+
+    def positions(self) -> List[Position]:
+        self._require_connected()
+        try:
+            raw = self._ib.positions()
+        except Exception as e:
+            raise RuntimeError(f"IBKR 持仓查询失败: {e}") from e
+
+        # Request market data for all contracts in batch to get current prices
+        _tick_map: dict = {}
+        try:
+            contracts = [p.contract for p in (raw or [])]
+            if contracts:
+                tickers = self._ib.reqTickers(*contracts)
+                for tk in tickers:
+                    _sym = tk.contract.symbol
+                    # Use last price, fall back to close, then mark price
+                    _px = tk.last or tk.close or tk.marketPrice() or 0.0
+                    if _px and _px > 0:
+                        _tick_map[_sym] = float(_px)
+        except Exception:
+            pass  # tick fetch best-effort; fall back to cost-based estimates below
+
+        result = []
+        for p in (raw or []):
+            contract = p.contract
+            sym  = contract.symbol
+            qty  = float(p.position)
+            cost = float(p.avgCost)
+            price = _tick_map.get(sym, 0.0)
+            # If tick fetch failed, estimate from unrealizedPNL + cost basis
+            if price == 0.0 and qty != 0 and cost > 0:
+                upnl = float(getattr(p, "unrealizedPNL", 0))
+                price = cost + upnl / qty
+                price = max(price, 0.0)
+            mv   = qty * price
+            pnl  = float(getattr(p, "unrealizedPNL", 0))
+            currency = contract.currency or "USD"
+            result.append(Position(
+                symbol=sym, name=contract.localSymbol or sym,
+                quantity=qty, available_qty=qty,
+                cost_price=cost, current_price=price,
+                market_value=mv, pnl=pnl,
+                pnl_pct=round(pnl / (cost * qty) * 100, 2) if cost and qty else 0.0,
+                currency=currency, market="us",
+            ))
+        return result
+
+    def orders(self, status: str = "all", limit: int = 50) -> List[Order]:
+        self._require_connected()
+        try:
+            raw = self._ib.openOrders() + self._ib.reqCompletedOrders(apiOnly=False)
+        except Exception as e:
+            raise RuntimeError(f"IBKR 订单查询失败: {e}") from e
+
+        result = []
+        for trade in list(raw or [])[:limit]:
+            order    = trade.order
+            contract = trade.contract
+            side     = "buy" if order.action.upper() == "BUY" else "sell"
+            mapped   = _ibkr_order_status(trade.orderStatus.status)
+            if status != "all" and mapped != status:
+                continue
+            result.append(Order(
+                order_id=str(order.orderId),
+                symbol=contract.symbol,
+                name=contract.localSymbol or contract.symbol,
+                side=side,
+                order_type=order.orderType.lower(),
+                quantity=float(order.totalQuantity),
+                filled_qty=float(trade.orderStatus.filled),
+                price=float(getattr(order, "lmtPrice", 0)),
+                avg_price=float(trade.orderStatus.avgFillPrice),
+                status=mapped,
+                currency=contract.currency or "USD",
+            ))
+        return result
+
+    def place_order(self, symbol: str, side: str, quantity: float,
+                    order_type: str = "limit", price: float = 0.0, **kwargs) -> OrderResult:
+        if self._readonly:
+            return OrderResult(success=False, message="IBKR 账户配置为只读模式", broker_id=self.broker_id)
+        self._require_connected()
+        try:
+            from ib_insync import Stock, MarketOrder, LimitOrder
+            contract = Stock(symbol, "SMART", "USD")
+            if order_type == "market":
+                order = MarketOrder(action=side.upper(), totalQuantity=int(quantity))
+            else:
+                order = LimitOrder(action=side.upper(), totalQuantity=int(quantity), lmtPrice=price)
+            trade = self._ib.placeOrder(contract, order)
+            return OrderResult(success=True, order_id=str(trade.order.orderId), broker_id=self.broker_id)
+        except Exception as e:
+            return OrderResult(success=False, message=str(e), broker_id=self.broker_id)
+
+    def cancel_order(self, order_id: str) -> bool:
+        self._require_connected()
+        try:
+            open_trades = self._ib.openTrades()
+            for t in open_trades:
+                if str(t.order.orderId) == order_id:
+                    self._ib.cancelOrder(t.order)
+                    return True
+            return False
+        except Exception:
+            return False
+
+    def _require_connected(self):
+        if not self._connected or not self._ib:
+            raise RuntimeError("IBKR 未连接，请先调用 connect()")
+
+
+def _ibkr_order_status(raw: str) -> str:
+    raw = raw.lower()
+    if raw == "filled":
+        return "filled"
+    if "partial" in raw:
+        return "partial"
+    if raw in ("cancelled", "inactive"):
+        return "cancelled"
+    return "open"

aria_code/brokers/intl/trading212_broker.py

@@ -0,0 +1,183 @@
+"""
+brokers/intl/trading212_broker.py — Trading212 (UK) 适配器
+============================================================
+支持市场：英国 / 欧洲股票 + ETF（Trading212 Invest/ISA 账户）
+
+无官方 SDK，直接用 requests 调 Trading212 公开 REST API。
+文档：https://t212public-api-docs.redoc.ly/
+
+⚠️ Trading212 的公开 API 目前只支持限价单（无市价单端点）——place_order
+   遇到 order_type="market" 会直接拒绝，不发请求，而不是静默改成限价单。
+
+配置示例::
+
+    {
+      "id":       "trading212_uk",
+      "type":     "trading212",
+      "label":    "Trading212 英国账户",
+      "api_key":  "xxx",
+      "base_url": "https://live.trading212.com/api/v0"
+    }
+
+practice/demo 账户把 base_url 换成 https://demo.trading212.com/api/v0。
+"""
+
+from __future__ import annotations
+
+from typing import Any, Dict, List
+
+from ..base import AccountInfo, BrokerBase, Order, OrderResult, Position
+
+DEFAULT_BASE_URL = "https://live.trading212.com/api/v0"
+
+
+class Trading212Broker(BrokerBase):
+    broker_type = "trading212"
+    broker_name = "Trading212"
+    market = "UK"
+
+    def __init__(self, broker_id: str, config: Dict[str, Any]):
+        super().__init__(broker_id, config)
+        self._api_key = config.get("api_key", "")
+        self._base_url = str(config.get("base_url") or DEFAULT_BASE_URL).rstrip("/")
+        self._session = None
+
+    def connect(self) -> bool:
+        try:
+            import requests
+        except ImportError:
+            raise ImportError("requests 未安装。请运行: pip install requests")
+
+        session = requests.Session()
+        session.headers.update({"Authorization": self._api_key})
+        try:
+            resp = session.get(f"{self._base_url}/equity/account/info", timeout=10)
+            resp.raise_for_status()
+            self._session = session
+            self._connected = True
+            return True
+        except Exception as e:
+            raise RuntimeError(f"Trading212 连接失败: {e}") from e
+
+    def _require_connected(self):
+        if not self._connected or not self._session:
+            raise RuntimeError("Trading212 未连接，请先调用 connect()")
+
+    def account_info(self) -> AccountInfo:
+        self._require_connected()
+        try:
+            info = self._session.get(f"{self._base_url}/equity/account/info", timeout=10).json()
+            cash = self._session.get(f"{self._base_url}/equity/account/cash", timeout=10).json()
+        except Exception as e:
+            raise RuntimeError(f"Trading212 账户查询失败: {e}") from e
+
+        free_cash = float(cash.get("free", 0) or 0)
+        total = float(cash.get("total", 0) or 0)
+        invested = float(cash.get("invested", 0) or 0)
+        ppl = float(cash.get("ppl", 0) or 0)
+        return AccountInfo(
+            broker_id=self.broker_id,
+            broker_type=self.broker_type,
+            label=self.label,
+            account_id=str(info.get("id", "")),
+            currency=str(info.get("currencyCode", "GBP")),
+            total_assets=total,
+            cash=free_cash,
+            market_value=invested,
+            pnl_today=ppl,
+            pnl_total=ppl,
+            extra={"blocked": float(cash.get("blocked", 0) or 0)},
+        )
+
+    def positions(self) -> List[Position]:
+        self._require_connected()
+        try:
+            raw = self._session.get(f"{self._base_url}/equity/portfolio", timeout=10).json()
+        except Exception as e:
+            raise RuntimeError(f"Trading212 持仓查询失败: {e}") from e
+
+        result = []
+        for p in (raw or []):
+            qty = float(p.get("quantity", 0) or 0)
+            cost = float(p.get("averagePrice", 0) or 0)
+            price = float(p.get("currentPrice", 0) or 0)
+            pnl = float(p.get("ppl", 0) or 0)
+            mv = qty * price
+            pnl_pct = (pnl / (mv - pnl) * 100) if (mv - pnl) else 0.0
+            ticker = str(p.get("ticker", ""))
+            result.append(Position(
+                symbol=ticker, name=ticker,
+                quantity=qty, available_qty=qty,
+                cost_price=cost, current_price=price,
+                market_value=mv, pnl=pnl, pnl_pct=pnl_pct,
+                currency="GBP", market="uk",
+            ))
+        return result
+
+    def orders(self, status: str = "all", limit: int = 50) -> List[Order]:
+        self._require_connected()
+        try:
+            raw = self._session.get(f"{self._base_url}/equity/orders", timeout=10).json()
+        except Exception as e:
+            raise RuntimeError(f"Trading212 订单查询失败: {e}") from e
+
+        result = []
+        for o in (raw or [])[:limit]:
+            mapped = _t212_order_status(str(o.get("status", "")))
+            if status not in ("all",) and mapped != status:
+                continue
+            result.append(Order(
+                order_id=str(o.get("id", "")),
+                symbol=str(o.get("ticker", "")),
+                name=str(o.get("ticker", "")),
+                side="buy" if float(o.get("quantity", 0) or 0) >= 0 else "sell",
+                order_type="limit",
+                quantity=abs(float(o.get("quantity", 0) or 0)),
+                filled_qty=abs(float(o.get("filledQuantity", 0) or 0)),
+                price=float(o.get("limitPrice", 0) or 0),
+                avg_price=float(o.get("fillPrice", 0) or 0),
+                status=mapped,
+                created_at=str(o.get("creationTime", "")),
+                currency="GBP",
+            ))
+        return result
+
+    def place_order(self, symbol: str, side: str, quantity: float,
+                    order_type: str = "limit", price: float = 0.0, **kwargs) -> OrderResult:
+        self._require_connected()
+        if order_type != "limit":
+            return OrderResult(success=False, message="Trading212 仅支持限价单", broker_id=self.broker_id)
+        if not price:
+            return OrderResult(success=False, message="限价单需要指定 price", broker_id=self.broker_id)
+        signed_qty = quantity if side == "buy" else -quantity
+        try:
+            resp = self._session.post(
+                f"{self._base_url}/equity/orders/limit",
+                json={"ticker": symbol, "quantity": signed_qty, "limitPrice": price, "timeValidity": "DAY"},
+                timeout=10,
+            )
+            resp.raise_for_status()
+            data = resp.json()
+            return OrderResult(success=True, order_id=str(data.get("id", "")), broker_id=self.broker_id)
+        except Exception as e:
+            return OrderResult(success=False, message=str(e), broker_id=self.broker_id)
+
+    def cancel_order(self, order_id: str) -> bool:
+        self._require_connected()
+        try:
+            resp = self._session.delete(f"{self._base_url}/equity/orders/{order_id}", timeout=10)
+            resp.raise_for_status()
+            return True
+        except Exception:
+            return False
+
+
+def _t212_order_status(raw: str) -> str:
+    raw = raw.upper()
+    if raw in ("FILLED",):
+        return "filled"
+    if raw in ("CANCELLED", "REJECTED", "EXPIRED"):
+        return "cancelled"
+    if raw in ("PARTIALLY_FILLED",):
+        return "partial"
+    return "open"

aria_code/brokers/intl/webull_broker.py

@@ -0,0 +1,156 @@
+"""
+brokers/intl/webull_broker.py — Webull 适配器
+==============================================
+支持市场：美股、港股（非官方 API，以查询为主）
+
+安装：pip install webull
+文档：https://github.com/tedchou12/webull
+
+配置示例::
+
+    {
+      "id":        "webull_us",
+      "type":      "webull",
+      "label":     "Webull 美股",
+      "username":  "your@email.com",
+      "password":  "your_password",
+      "device_id": "",
+      "mfa":       ""
+    }
+
+注意：Webull 为非官方 API，下单功能可能不稳定。
+建议仅用于查询（持仓/账户），下单在 App 内操作。
+"""
+
+from __future__ import annotations
+
+from typing import Any, Dict, List
+
+from ..base import BrokerBase, AccountInfo, Position, Order, OrderResult
+
+
+class WebullBroker(BrokerBase):
+    broker_type = "webull"
+    broker_name = "Webull"
+    market      = "US"
+    read_only   = True          # 默认只读，避免非官方 API 下单风险
+
+    def __init__(self, broker_id: str, config: Dict[str, Any]):
+        super().__init__(broker_id, config)
+        self._username  = config.get("username",  "")
+        self._password  = config.get("password",  "")
+        self._device_id = config.get("device_id", "")
+        self._mfa       = config.get("mfa",        "")
+        self._wb        = None
+
+    def connect(self) -> bool:
+        try:
+            from webull import webull as wb_cls
+        except ImportError:
+            raise ImportError("webull 未安装。请运行: pip install webull")
+
+        try:
+            wb = wb_cls()
+            if self._device_id:
+                wb._set_did(self._device_id)
+            wb.login(self._username, self._password, mfa_code=self._mfa or None)
+            self._wb        = wb
+            self._connected = True
+            return True
+        except Exception as e:
+            raise RuntimeError(f"Webull 登录失败: {e}") from e
+
+    def account_info(self) -> AccountInfo:
+        self._require_connected()
+        try:
+            data = self._wb.get_account()
+        except Exception as e:
+            raise RuntimeError(f"Webull 账户查询失败: {e}") from e
+
+        total   = float(data.get("netLiquidation",  data.get("totalMarketValue", 0)))
+        cash    = float(data.get("cashBalance",     data.get("availableBalance", 0)))
+        mv      = float(data.get("totalMarketValue", 0))
+        pnl_day = float(data.get("unrealizedProfitLoss", 0))
+        return AccountInfo(
+            broker_id=self.broker_id,
+            broker_type=self.broker_type,
+            label=self.label,
+            account_id=str(data.get("accountId", "")),
+            currency="USD",
+            total_assets=total,
+            cash=cash,
+            market_value=mv,
+            pnl_today=pnl_day,
+        )
+
+    def positions(self) -> List[Position]:
+        self._require_connected()
+        try:
+            raw = self._wb.get_positions()
+        except Exception as e:
+            raise RuntimeError(f"Webull 持仓查询失败: {e}") from e
+
+        result = []
+        for p in (raw or []):
+            ticker = p.get("ticker", {})
+            sym    = str(ticker.get("symbol",    p.get("symbol", "")))
+            name   = str(ticker.get("name",      ""))
+            qty    = float(p.get("position",     p.get("quantity", 0)))
+            cost   = float(p.get("costPrice",    p.get("avgCost",  0)))
+            price  = float(p.get("lastPrice",    p.get("price",    0)))
+            mv     = float(p.get("marketValue",  qty * price))
+            pnl    = float(p.get("unrealizedProfitLoss", 0))
+            pnl_pct= float(p.get("unrealizedProfitLossRate", 0)) * 100
+            result.append(Position(
+                symbol=sym, name=name,
+                quantity=qty, available_qty=qty,
+                cost_price=cost, current_price=price,
+                market_value=mv, pnl=pnl, pnl_pct=pnl_pct,
+                currency="USD", market="us",
+            ))
+        return result
+
+    def orders(self, status: str = "all", limit: int = 50) -> List[Order]:
+        self._require_connected()
+        try:
+            raw = self._wb.get_history_orders(status="All")
+        except Exception as e:
+            raise RuntimeError(f"Webull 订单查询失败: {e}") from e
+
+        result = []
+        for o in list(raw or [])[:limit]:
+            side   = "buy" if str(o.get("action", "")).upper() == "BUY" else "sell"
+            mapped = _wb_order_status(str(o.get("status", "")))
+            if status != "all" and mapped != status:
+                continue
+            ticker = o.get("ticker", {})
+            result.append(Order(
+                order_id=str(o.get("orderId", "")),
+                symbol=str(ticker.get("symbol", o.get("symbol", ""))),
+                name=str(ticker.get("name", "")),
+                side=side,
+                order_type=str(o.get("orderType", "limit")).lower(),
+                quantity=float(o.get("totalQuantity", 0)),
+                filled_qty=float(o.get("filledQuantity", 0)),
+                price=float(o.get("lmtPrice", 0) or 0),
+                avg_price=float(o.get("avgFilledPrice", 0) or 0),
+                status=mapped,
+                created_at=str(o.get("createTime", "")),
+                currency="USD",
+            ))
+        return result
+
+    def _require_connected(self):
+        if not self._connected or not self._wb:
+            raise RuntimeError("Webull 未连接，请先调用 connect()")
+
+
+def _wb_order_status(raw: str) -> str:
+    raw = raw.upper()
+    if raw == "FILLED":
+        return "filled"
+    if raw == "PARTIALLY_FILLED":
+        return "partial"
+    if raw in ("CANCELLED", "EXPIRED", "REJECTED"):
+        return "cancelled"
+    return "open"