fix ipo bug

py-client/sdk/trade.py

@@ -8,6 +8,64 @@ PR_TYPE_LATEST = 5
 QUICK_TRADE_NOW = 2
 ORDER_SIDE_BY_OFFSET = {"23": "BUY", "24": "SELL", "48": "BUY", "49": "SELL"}
 
+# QMT 的发行数据按市场再分一层时使用的市场键。
+_IPO_MARKETS = {"SH", "SZ", "BJ"}
+
+
+def _ipo_stock_code(code: Any, info: dict[str, Any]) -> str:
+    """把 QMT 的证券代码键补成全码 ``600000.SH``。"""
+    text = str(code or "").strip()
+    if not text or "." in text:
+        return text
+    market = str(
+        info.get("market") or info.get("exchange") or info.get("ExchangeID") or ""
+    ).strip().upper()
+    return f"{text}.{market}" if market else text
+
+
+def _ipo_entry(code: Any, info: dict[str, Any]) -> dict[str, Any]:
+    """保留原始发行信息，并补出候选列表使用的 ``stock`` 字段。"""
+    entry = dict(info)
+    stock = _ipo_stock_code(code, entry)
+    if stock:
+        entry["stock"] = stock
+    return entry
+
+
+def _ipo_candidates(response: Any) -> list[dict[str, Any]]:
+    """把 ``/api/trade/ipo_data`` 的响应规范化为候选列表。
+
+    QMT 的 ``get_ipo_data(type)`` 返回 ``{证券代码: 发行信息}`` 字典，部分
+    版本再按市场分一层 ``{市场: {证券代码: 发行信息}}``，旧版服务端还会包
+    一层 ``{"data": ...}``。空响应表示当日没有可申购标的；无法识别的结构抛
+    ``ValueError``，避免把接口异常静默当成“今日无新股”。
+    """
+    if isinstance(response, dict) and len(response) == 1 and "data" in response:
+        response = response["data"]
+    if response is None:
+        return []
+    if isinstance(response, list):
+        # 列表逐项交给策略层校验，单条异常不影响其他候选。
+        return list(response)
+    if not isinstance(response, dict):
+        raise ValueError(f"unsupported IPO response type: {type(response).__name__}")
+    if not response:
+        return []
+    if not all(isinstance(value, dict) for value in response.values()):
+        raise ValueError("IPO response values must be objects")
+
+    candidates: list[dict[str, Any]] = []
+    for key, value in response.items():
+        market = str(key).strip().upper()
+        if market in _IPO_MARKETS and all(isinstance(item, dict) for item in value.values()):
+            for code, info in value.items():
+                entry = dict(info)
+                entry.setdefault("market", market)
+                candidates.append(_ipo_entry(code, entry))
+        else:
+            candidates.append(_ipo_entry(key, value))
+    return candidates
+
 
 class TradeMixin:
     def passorder(
@@ -43,14 +101,15 @@ class TradeMixin:
         )
 
     def ipo_data(self, ipo_type: str = "STOCK") -> list[dict[str, Any]]:
-        """Return today's IPO candidates from the QMT REST service."""
+        """Return today's IPO candidates from the QMT REST service.
+
+        QMT 按证券代码返回字典，这里统一成候选列表，字段名保持不变。
+        """
         response = self._post_json(
             "/api/trade/ipo_data",
             {"type": str(ipo_type).strip().upper()},
         )
-        if not isinstance(response, list):
-            raise ValueError('IPO response must be a list')
-        return response
+        return _ipo_candidates(response)
 
 
     def cancel_by_id(self, order_id: str) -> dict[str, Any]: