aria-code 0.52.0__py3-none-any.whl

aria_code/brokers/config.py

@@ -0,0 +1,414 @@
+"""
+brokers/config.py — 券商配置加载与管理
+========================================
+配置文件位置：~/.aria-code/brokers.json（或由 ARIA_HOME 环境变量覆盖）
+
+示例配置::
+
+    {
+      "brokers": [
+        {
+          "id":         "xt_main",
+          "type":       "xtquant",
+          "label":      "中信主账户",
+          "account_id": "1234567890",
+          "default":    true
+        },
+        {
+          "id":         "ibkr_us",
+          "type":       "ibkr",
+          "label":      "盈透美股",
+          "host":       "127.0.0.1",
+          "port":       7496,
+          "client_id":  1
+        },
+        {
+          "id":         "alpaca_paper",
+          "type":       "alpaca",
+          "label":      "Alpaca 模拟盘",
+          "api_key":    "PKxxx",
+          "api_secret": "xxx",
+          "paper":      true
+        },
+        {
+          "id":         "futu_hk",
+          "type":       "futu",
+          "label":      "富途港股",
+          "host":       "127.0.0.1",
+          "port":       11111,
+          "market":     "HK"
+        },
+        {
+          "id":         "tiger_us",
+          "type":       "tiger",
+          "label":      "老虎美股",
+          "tiger_id":   "xxx",
+          "private_key_path": "~/.aria-code/tiger_rsa.pem",
+          "account":    "xxx"
+        },
+        {
+          "id":         "longbridge_cn",
+          "type":       "longbridge",
+          "label":      "长桥A股",
+          "app_key":    "xxx",
+          "app_secret": "xxx",
+          "access_token": "xxx"
+        }
+      ]
+    }
+
+支持的 type 值:
+    paper       本地仿盘账户（无真实券商连接）
+    xtquant     迅投（中信、华鑫、浙商等）
+    easytrader  easytrader（同花顺、通达信、华泰、国君）
+    futu        富途牛牛 OpenAPI（港股/美股/A股）
+    tiger       老虎证券 OpenAPI（美股/港股/A股）
+    longbridge  长桥证券 OpenAPI（港股/美股/A股）
+    ibkr        Interactive Brokers TWS/Gateway
+    alpaca      Alpaca Markets（美股，支持模拟盘）
+    webull      Webull（美股）
+"""
+
+from __future__ import annotations
+
+import json
+import os
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+from aria_code.packages.aria_core.paths import aria_home
+
+
+def _resolve_aria_home() -> Path:
+    """Resolve the Aria config directory (same logic as aria_cli.py).
+
+    Priority:
+      1. ARIA_HOME env var (explicit override)
+      2. ~/.arthera — legacy path, kept for backward compat if it exists
+      3. ~/.aria-code — default for fresh installs
+    """
+    if "ARIA_HOME" in os.environ:
+        return Path(os.environ["ARIA_HOME"]).expanduser()
+    legacy = aria_home()
+    if legacy.exists():
+        return legacy
+    return Path.home() / ".aria-code"
+
+
+BROKERS_CONFIG_PATH = _resolve_aria_home() / "brokers.json"
+
+
+# ── 读写工具 ───────────────────────────────────────────────────────────────────
+
+def load_config() -> Dict[str, Any]:
+    """加载 brokers.json，文件不存在时返回空 {"brokers": []}。"""
+    if not BROKERS_CONFIG_PATH.exists():
+        return {"brokers": []}
+    try:
+        with open(BROKERS_CONFIG_PATH, encoding="utf-8") as f:
+            return json.load(f)
+    except Exception:
+        return {"brokers": []}
+
+
+def save_config(cfg: Dict[str, Any]) -> None:
+    """保存配置到 brokers.json（0600，仅本人可读）。
+
+    这个文件存的是券商密码 / api_secret / access_token —— 本仓库里价值最高的
+    凭据。此前它按进程 umask 写出，通常是 0644，同机器上任何账号可读。
+    """
+    from aria_code.packages.aria_core.secure_file import write_secret_json
+
+    write_secret_json(BROKERS_CONFIG_PATH, cfg)
+
+
+def list_broker_configs() -> List[Dict[str, Any]]:
+    """返回所有已配置的券商列表。"""
+    return load_config().get("brokers", [])
+
+
+def get_broker_config(broker_id: str) -> Optional[Dict[str, Any]]:
+    """根据 id 查找某个券商配置。"""
+    for b in list_broker_configs():
+        if b.get("id") == broker_id:
+            return b
+    return None
+
+
+def get_default_broker_config() -> Optional[Dict[str, Any]]:
+    """返回标记了 default=true 的券商，或第一个。"""
+    brokers = list_broker_configs()
+    if not brokers:
+        return None
+    for b in brokers:
+        if b.get("default"):
+            return b
+    return brokers[0]
+
+
+def add_broker_config(cfg: Dict[str, Any]) -> None:
+    """新增一条券商配置（id 重复时覆盖）。"""
+    data = load_config()
+    brokers = data.get("brokers", [])
+    brokers = [b for b in brokers if b.get("id") != cfg.get("id")]
+    brokers.append(cfg)
+    data["brokers"] = brokers
+    save_config(data)
+
+
+def remove_broker_config(broker_id: str) -> bool:
+    """删除某个券商配置，返回 True 表示找到并删除。"""
+    data = load_config()
+    before = len(data.get("brokers", []))
+    data["brokers"] = [b for b in data.get("brokers", []) if b.get("id") != broker_id]
+    if len(data["brokers"]) < before:
+        save_config(data)
+        return True
+    return False
+
+
+def set_default_broker(broker_id: str) -> bool:
+    """设置默认券商。"""
+    data = load_config()
+    found = False
+    for b in data.get("brokers", []):
+        if b.get("id") == broker_id:
+            b["default"] = True
+            found = True
+        else:
+            b.pop("default", None)
+    if found:
+        save_config(data)
+    return found
+
+
+# ── 聊天内确认下单开关 ──────────────────────────────────────────────────────────
+# 默认关闭，且只能在终端里手动开启（见 apps/cli/commands/broker_cmds.py 的
+# `/trade allow-chat-confirm <broker_id>`——那条命令本身要求真人在终端里
+# 原样重新输入这个券商的 id（不是 y/yes）才会调用 set_chat_confirm_enabled）。
+# 这个开关本身永远不能通过 MCP/聊天打开，只能被查询和使用——打开这道门的
+# 动作必须发生在终端键盘前，跟 /trade confirm 那道"真正下单"的门是同一个
+# 安全等级。见 packages/aria_mcp/server.py 的 _call_broker_confirm_order。
+
+def is_chat_confirm_enabled(broker_id: str) -> bool:
+    """某个券商是否已经被用户在终端里显式开启了'聊天内确认下单'。"""
+    cfg = get_broker_config(broker_id)
+    return bool(cfg and cfg.get("chat_confirm_enabled") is True)
+
+
+def set_chat_confirm_enabled(broker_id: str, enabled: bool) -> bool:
+    """开启/关闭某个券商的'聊天内确认下单'。返回 True 表示找到并设置成功。
+
+    调用方（`/broker allow-chat-confirm`）负责在调用这个函数之前，先让
+    真人在终端里做一次明确、不容易误触的确认——这个函数本身不做任何
+    额外确认，只是纯粹的配置读写。"""
+    data = load_config()
+    found = False
+    for b in data.get("brokers", []):
+        if b.get("id") == broker_id:
+            b["chat_confirm_enabled"] = bool(enabled)
+            found = True
+            break
+    if found:
+        save_config(data)
+    return found
+
+
+# ── 配置校验 ──────────────────────────────────────────────────────────────────
+
+# 每种 type 必需的字段
+_REQUIRED_FIELDS: Dict[str, List[str]] = {
+    "paper":      [],
+    "xtquant":    ["account_id"],
+    "easytrader": ["broker_name"],
+    "futu":       ["host", "port"],           # FutuOpenD must be running locally
+    "tiger":      ["tiger_id", "private_key_path", "account"],
+    "longbridge": ["app_key", "app_secret", "access_token"],
+    "ibkr":       ["host", "port"],           # TWS/Gateway host:port required
+    "alpaca":     ["api_key", "api_secret"],
+    "webull":     ["username", "password"],   # read-only, still needs credentials
+    "trading212": ["api_key"],
+}
+
+_TYPE_LABELS: Dict[str, str] = {
+    "paper":      "Aria 本地仿盘账户",
+    "xtquant":    "迅投 XTQuant（中信/华鑫/浙商等）",
+    "easytrader": "EasyTrader（同花顺/通达信/华泰/国君）",
+    "futu":       "富途牛牛 OpenAPI",
+    "tiger":      "老虎证券 OpenAPI",
+    "longbridge": "长桥证券 OpenAPI",
+    "ibkr":       "Interactive Brokers TWS/Gateway",
+    "alpaca":     "Alpaca Markets（美股 / 模拟盘）",
+    "webull":     "Webull（美股）",
+    "trading212": "Trading212（英国 Invest/ISA）",
+}
+
+
+_RECOMMENDED_FIELDS: Dict[str, List[str]] = {
+    "paper":  ["starting_cash", "currency"],
+    "futu":   ["market"],
+    "ibkr":   ["client_id"],
+    "alpaca": ["paper"],
+}
+
+
+def validate_broker_config(cfg: Dict[str, Any]) -> List[str]:
+    """校验券商配置，返回错误列表（空列表=通过）。
+
+    Warnings (prefixed with '⚠ ') are non-fatal but worth showing.
+    """
+    errors: List[str] = []
+    if not cfg.get("id"):
+        errors.append("缺少 'id' 字段")
+    if not cfg.get("label"):
+        errors.append("⚠ 建议设置 'label' 字段（账户显示名称）")
+    broker_type = cfg.get("type", "")
+    if not broker_type:
+        errors.append("缺少 'type' 字段")
+    elif broker_type not in _REQUIRED_FIELDS:
+        errors.append(f"不支持的 type: {broker_type!r}  (支持: {', '.join(_REQUIRED_FIELDS)})")
+    else:
+        for req in _REQUIRED_FIELDS[broker_type]:
+            if not cfg.get(req):
+                errors.append(f"{broker_type} 缺少必需字段: '{req}'")
+        for rec in _RECOMMENDED_FIELDS.get(broker_type, []):
+            if not cfg.get(rec):
+                errors.append(f"⚠ {broker_type} 建议设置 '{rec}' 字段")
+    return errors
+
+
+def supported_broker_types() -> Dict[str, str]:
+    """返回所有支持的券商类型 {type: 描述}。"""
+    return dict(_TYPE_LABELS)
+
+
+# ── 生成配置模板 ──────────────────────────────────────────────────────────────
+
+_TEMPLATES: Dict[str, Dict[str, Any]] = {
+    "paper": {
+        "id": "paper_main",
+        "type": "paper",
+        "label": "Aria 仿盘账户",
+        "mode": "paper",
+        "starting_cash": 100000,
+        "currency": "USD",
+        "default": True,
+        "_comment": "本地仿盘账户，不连接真实券商；订单写入 ~/.arthera/paper_ledger.json"
+    },
+    "xtquant": {
+        "id": "xt_main",
+        "type": "xtquant",
+        "label": "中信主账户",
+        "mode": "read_only",
+        "allow_live_trade": False,
+        "require_confirm": True,
+        "account_id": "YOUR_ACCOUNT_ID",
+        "default": True,
+        "_comment": "需安装 xtquant: pip install xtquant  (仅 Windows/Linux)"
+    },
+    "easytrader": {
+        "id": "ht_main",
+        "type": "easytrader",
+        "label": "华泰账户",
+        "mode": "read_only",
+        "allow_live_trade": False,
+        "require_confirm": True,
+        "broker_name": "huatai",
+        "exe_path": "C:\\华泰证券\\xiadan.exe",
+        "_comment": "broker_name 可选: huatai/guojun/ths/tdx/yh/zszq/xq"
+    },
+    "futu": {
+        "id": "futu_hk",
+        "type": "futu",
+        "label": "富途港股",
+        "mode": "read_only",
+        "allow_live_trade": False,
+        "require_confirm": True,
+        "host": "127.0.0.1",
+        "port": 11111,
+        "market": "HK",
+        "_comment": "需安装 futu-api: pip install futu-api  并启动富途 OpenD"
+    },
+    "tiger": {
+        "id": "tiger_us",
+        "type": "tiger",
+        "label": "老虎美股",
+        "mode": "read_only",
+        "allow_live_trade": False,
+        "require_confirm": True,
+        "tiger_id": "YOUR_TIGER_ID",
+        "private_key_path": "~/.aria-code/tiger_rsa.pem",
+        "account": "YOUR_ACCOUNT",
+        "_comment": "需安装 tigeropen: pip install tigeropen"
+    },
+    "longbridge": {
+        "id": "lb_cn",
+        "type": "longbridge",
+        "label": "长桥A股",
+        "mode": "read_only",
+        "allow_live_trade": False,
+        "require_confirm": True,
+        "app_key": "YOUR_APP_KEY",
+        "app_secret": "YOUR_APP_SECRET",
+        "access_token": "YOUR_ACCESS_TOKEN",
+        "_comment": "需安装 longbridge: pip install longbridge"
+    },
+    "ibkr": {
+        "id": "ibkr_us",
+        "type": "ibkr",
+        "label": "盈透美股",
+        "mode": "read_only",
+        "allow_live_trade": False,
+        "require_confirm": True,
+        "host": "127.0.0.1",
+        "port": 7496,
+        "client_id": 1,
+        "_comment": "需安装 ib_insync: pip install ib_insync  并启动 TWS/Gateway"
+    },
+    "alpaca": {
+        "id": "alpaca_paper",
+        "type": "alpaca",
+        "label": "Alpaca 模拟盘",
+        "mode": "paper",
+        "allow_live_trade": False,
+        "require_confirm": True,
+        "api_key": "YOUR_API_KEY",
+        "api_secret": "YOUR_API_SECRET",
+        "paper": True,
+        "_comment": "需安装 alpaca-py: pip install alpaca-py"
+    },
+    "webull": {
+        "id": "webull_us",
+        "type": "webull",
+        "label": "Webull 美股",
+        "mode": "read_only",
+        "allow_live_trade": False,
+        "require_confirm": True,
+        "username": "YOUR_EMAIL_OR_PHONE",
+        "password": "YOUR_PASSWORD",
+        "device_id": "",
+        "_comment": "需安装 webull: pip install webull"
+    },
+    "trading212": {
+        "id": "trading212_uk",
+        "type": "trading212",
+        "label": "Trading212 英国账户",
+        "mode": "read_only",
+        "allow_live_trade": False,
+        "require_confirm": True,
+        "api_key": "YOUR_API_KEY",
+        "base_url": "https://live.trading212.com/api/v0",
+        "_comment": "无需额外 SDK（直接 REST）。practice 账户把 base_url 换成 https://demo.trading212.com/api/v0。仅支持限价单。"
+    },
+}
+
+
+def get_config_template(broker_type: str) -> Optional[Dict[str, Any]]:
+    """返回指定券商类型的配置模板。"""
+    import copy
+    tmpl = _TEMPLATES.get(broker_type)
+    return copy.deepcopy(tmpl) if tmpl else None
+
+
+def print_all_templates() -> str:
+    """返回所有配置模板的 JSON 字符串（用于 /broker init 命令）。"""
+    return json.dumps({"brokers": list(_TEMPLATES.values())}, ensure_ascii=False, indent=2)

aria_code/brokers/intl/__init__.py

@@ -0,0 +1,5 @@
+from .ibkr_broker   import IBKRBroker
+from .alpaca_broker import AlpacaBroker
+from .webull_broker import WebullBroker
+
+__all__ = ["IBKRBroker", "AlpacaBroker", "WebullBroker"]

aria_code/brokers/intl/alpaca_broker.py

@@ -0,0 +1,183 @@
+"""
+brokers/intl/alpaca_broker.py — Alpaca Markets 适配器
+======================================================
+支持市场：美股（实盘 + 模拟盘）
+
+安装：pip install alpaca-py
+文档：https://docs.alpaca.markets/
+
+配置示例::
+
+    {
+      "id":         "alpaca_paper",
+      "type":       "alpaca",
+      "label":      "Alpaca 模拟盘",
+      "api_key":    "PKxxx",
+      "api_secret": "xxx",
+      "paper":      true
+    }
+"""
+
+from __future__ import annotations
+
+from typing import Any, Dict, List
+
+from ..base import BrokerBase, AccountInfo, Position, Order, OrderResult
+
+
+class AlpacaBroker(BrokerBase):
+    broker_type = "alpaca"
+    broker_name = "Alpaca Markets"
+    market      = "US"
+
+    def __init__(self, broker_id: str, config: Dict[str, Any]):
+        super().__init__(broker_id, config)
+        self._api_key    = config.get("api_key",    "")
+        self._api_secret = config.get("api_secret", "")
+        self._paper      = bool(config.get("paper", False))
+        self._trading    = None
+
+    def connect(self) -> bool:
+        try:
+            from alpaca.trading.client import TradingClient
+        except ImportError:
+            raise ImportError("alpaca-py 未安装。请运行: pip install alpaca-py")
+
+        try:
+            client = TradingClient(self._api_key, self._api_secret, paper=self._paper)
+            client.get_account()          # 测试连通性
+            self._trading   = client
+            self._connected = True
+            return True
+        except Exception as e:
+            raise RuntimeError(f"Alpaca 连接失败: {e}") from e
+
+    def account_info(self) -> AccountInfo:
+        self._require_connected()
+        try:
+            a = self._trading.get_account()
+        except Exception as e:
+            raise RuntimeError(f"Alpaca 账户查询失败: {e}") from e
+
+        equity = float(getattr(a, "equity",        0))
+        cash   = float(getattr(a, "cash",           0))
+        mv     = float(getattr(a, "long_market_value", 0)) - float(getattr(a, "short_market_value", 0))
+        pnl    = float(getattr(a, "unrealized_pl",  0))
+        return AccountInfo(
+            broker_id=self.broker_id,
+            broker_type=self.broker_type,
+            label=self.label,
+            account_id=str(getattr(a, "account_number", getattr(a, "id", ""))),
+            currency="USD",
+            total_assets=equity,
+            cash=cash,
+            market_value=mv,
+            pnl_today=float(getattr(a, "unrealized_pl", 0)),
+            pnl_total=pnl,
+            extra={"buying_power": float(getattr(a, "buying_power", 0)),
+                   "paper": self._paper},
+        )
+
+    def positions(self) -> List[Position]:
+        self._require_connected()
+        try:
+            raw = self._trading.get_all_positions()
+        except Exception as e:
+            raise RuntimeError(f"Alpaca 持仓查询失败: {e}") from e
+
+        result = []
+        for p in (raw or []):
+            sym   = str(getattr(p, "symbol", ""))
+            qty   = float(getattr(p, "qty",             0))
+            avail = float(getattr(p, "qty_available",   qty))
+            cost  = float(getattr(p, "avg_entry_price", 0))
+            price = float(getattr(p, "current_price",   0))
+            mv    = float(getattr(p, "market_value",    0))
+            pnl   = float(getattr(p, "unrealized_pl",   0))
+            pnl_pct = float(getattr(p, "unrealized_plpc", 0)) * 100
+            result.append(Position(
+                symbol=sym, name=sym,
+                quantity=qty, available_qty=avail,
+                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:
+            from alpaca.trading.requests import GetOrdersRequest
+            from alpaca.trading.enums import QueryOrderStatus
+            qstatus = {
+                "open":      QueryOrderStatus.OPEN,
+                "filled":    QueryOrderStatus.CLOSED,
+                "cancelled": QueryOrderStatus.CLOSED,
+            }.get(status, QueryOrderStatus.ALL)
+            raw = self._trading.get_orders(GetOrdersRequest(status=qstatus, limit=limit))
+        except Exception as e:
+            raise RuntimeError(f"Alpaca 订单查询失败: {e}") from e
+
+        result = []
+        for o in (raw or []):
+            side   = "buy" if str(getattr(o, "side", "")).lower() == "buy" else "sell"
+            mapped = _alpaca_order_status(str(getattr(o, "status", "")))
+            if status not in ("all",) and mapped != status:
+                continue
+            result.append(Order(
+                order_id=str(getattr(o, "id", "")),
+                symbol=str(getattr(o, "symbol", "")),
+                name=str(getattr(o, "symbol", "")),
+                side=side,
+                order_type=str(getattr(o, "order_type", "limit")).lower(),
+                quantity=float(getattr(o, "qty",         0) or 0),
+                filled_qty=float(getattr(o, "filled_qty", 0) or 0),
+                price=float(getattr(o, "limit_price",    0) or 0),
+                avg_price=float(getattr(o, "filled_avg_price", 0) or 0),
+                status=mapped,
+                created_at=str(getattr(o, "created_at", "")),
+                currency="USD",
+            ))
+        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()
+        try:
+            from alpaca.trading.requests import MarketOrderRequest, LimitOrderRequest
+            from alpaca.trading.enums import OrderSide, TimeInForce
+            alpaca_side = OrderSide.BUY if side == "buy" else OrderSide.SELL
+            if order_type == "market":
+                req = MarketOrderRequest(
+                    symbol=symbol, qty=quantity, side=alpaca_side, time_in_force=TimeInForce.DAY)
+            else:
+                req = LimitOrderRequest(
+                    symbol=symbol, qty=quantity, side=alpaca_side,
+                    limit_price=price, time_in_force=TimeInForce.DAY)
+            o = self._trading.submit_order(req)
+            return OrderResult(success=True, order_id=str(o.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:
+            self._trading.cancel_order_by_id(order_id)
+            return True
+        except Exception:
+            return False
+
+    def _require_connected(self):
+        if not self._connected or not self._trading:
+            raise RuntimeError("Alpaca 未连接，请先调用 connect()")
+
+
+def _alpaca_order_status(raw: str) -> str:
+    raw = raw.lower()
+    if raw == "filled":
+        return "filled"
+    if raw == "partially_filled":
+        return "partial"
+    if raw in ("canceled", "cancelled", "expired", "rejected"):
+        return "cancelled"
+    return "open"