fix bug

py-client/sdk/portfolio.py

@@ -0,0 +1,52 @@
+from __future__ import annotations
+
+from typing import Any
+
+from .models import Assets, OrderItem, Portfolio, PositionItem
+
+
+class PortfolioMixin:
+    def portfolio(self) -> Portfolio:
+        data = self._get_json("/api/portfolio") or {}
+        positions = {
+            code: PositionItem.from_dict(value, code)
+            for code, value in data.get("positions", {}).items()
+        }
+        return Portfolio(
+            assets=Assets.from_dict(data.get("assets", {})),
+            positions=positions,
+            orders=[OrderItem.from_trade_detail(row) for row in data.get("orders", [])],
+        )
+
+    def positions(self) -> tuple[list[str], list[PositionItem]]:
+        data = self._get_json("/api/portfolio/positions") or {}
+        positions = [
+            PositionItem.from_dict(value, code)
+            for code, value in data.get("data", {}).items()
+        ]
+        return [item.stock_code for item in positions], positions
+
+    def assets(self) -> Assets:
+        return Assets.from_dict(self._get_json("/api/portfolio/assets") or {})
+
+    def orders(self) -> list[OrderItem]:
+        data = self._get_json("/api/portfolio/order") or []
+        return [OrderItem.from_trade_detail(row) for row in data]
+
+    def deals(self) -> list[dict[str, Any]]:
+        data = self._get_json("/api/portfolio/deal") or {}
+        return data.get("deals", [])
+
+    def trade_detail_data(self, datatype: str) -> Any:
+        datatype = str(datatype).strip().lower()
+        handlers = {
+            "account": self.assets,
+            "position": lambda: self.positions()[1],
+            "order": self.orders,
+            "deal": self.deals,
+        }
+        handler = handlers.get(datatype)
+        if handler is None:
+            raise ValueError(f"unsupported trade detail datatype: {datatype}")
+        result = handler()
+        return [result] if datatype == "account" else result