fix bug

py-client/sdk/v2.py

@@ -0,0 +1,86 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any
+from urllib.parse import urlencode
+
+from .client import Client as HTTPClient
+from .models import Assets, OrderItem, PositionItem
+
+
+@dataclass(slots=True)
+class Portfolio:
+    assets: Assets
+    positions: list[PositionItem]
+    orders: list[OrderItem]
+
+
+class Client_V2(HTTPClient):
+    """QMT ``/api/v2`` synchronous client."""
+
+    def portfolio(self) -> Portfolio:
+        data = self._get_json("/api/v2/portfolio", ) or {}
+        positions = data.get("positions", {})
+        orders = data.get("orders", {})
+        return Portfolio(
+            assets=Assets.from_dict(data.get("assets", {})),
+            positions=[PositionItem.from_trade_detail(row) for row in positions],
+            orders=[OrderItem.from_trade_detail(value) for value in orders],
+        )
+
+    def positions(self) -> list[PositionItem]:
+        data = self._post("/api/v2/positions", {"account": self.account_type}) or {}
+        return [
+            PositionItem.from_dict(value, code)
+            for code, value in data.get("data", {}).items()
+        ]
+
+    def assets(self) -> dict[str, Any]:
+        return self._post("/api/v2/assets", {"account": self.account_type}) or {}
+
+    def context_info(self) -> dict[str, Any]:
+        return self._get("/api/v2/context/info") or {}
+
+    def stock_name(self, stock_code: str) -> Any:
+        return self._get_ref("stock_name", stock_code)
+
+    def open_date(self, stock_code: str) -> Any:
+        return self._get_ref("open_date", stock_code)
+
+    def last_volume(self, stock_code: str) -> Any:
+        return self._get_ref("last_volume", stock_code)
+
+    def total_share(self, stock_code: str) -> Any:
+        return self._get_ref("total_share", stock_code)
+
+    def svol(self, stock_code: str) -> Any:
+        return self._get_ref("svol", stock_code)
+
+    def bvol(self, stock_code: str) -> Any:
+        return self._get_ref("bvol", stock_code)
+
+    def divid_factors(self, stock_code: str) -> Any:
+        return self._get_ref("divid_factors", stock_code)
+
+    def etf_info(self, stock_code: str) -> Any:
+        return self._get_ref("etf_info", stock_code)
+
+    def etf_iopv(self, stock_code: str) -> Any:
+        return self._get_ref("etf_iopv", stock_code)
+
+    def instrument_detail(self, stock_code: str) -> Any:
+        return self._get_ref("instrumentdetail", stock_code)
+
+    def his_st_data(self, stock_code: str) -> Any:
+        return self._get_ref("his_st_data", stock_code)
+
+    def _get_ref(self, endpoint: str, stock_code: str) -> Any:
+        query = urlencode({"stock_code": stock_code})
+        payload = self._get(f"/api/v2/get/{endpoint}?{query}") or {}
+        return payload.get("ref")
+
+
+Client = Client_V2
+
+
+__all__ = ["Client", "Client_V2", "Portfolio"]