feat

py-client/sdk/models.py

@@ -0,0 +1,106 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+from typing import Any
+
+
+def _number(value: Any, kind: type = float) -> Any:
+    try:
+        return kind(value or 0)
+    except (TypeError, ValueError):
+        return kind()
+
+
+@dataclass
+class Position:
+    stock_code: str = ""
+    stock_name: str = ""
+    direction: Any = None
+    volume: int = 0
+    open_price: float = 0.0
+    float_profit: float = 0.0
+    market_value: float = 0.0
+    stock_holder: str = ""
+    frozen_volume: int = 0
+    can_use_volume: int = 0
+    on_road_volume: int = 0
+    yesterday_volume: int = 0
+    last_price: float = 0.0
+    profit_rate: float = 0.0
+    future_trade_type: Any = None
+    expire_date: str = ""
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any], code: str = "") -> "Position":
+        return cls(
+            stock_code=str(data.get("StockCode") or code), stock_name=str(data.get("StockName") or ""),
+            direction=data.get("Direction"), volume=_number(data.get("Volume"), int),
+            open_price=_number(data.get("OpenPrice")), float_profit=_number(data.get("FloatProfit")),
+            market_value=_number(data.get("MarketValue")), stock_holder=str(data.get("StockHolder") or ""),
+            frozen_volume=_number(data.get("FrozenVolume"), int), can_use_volume=_number(data.get("CanUseVolume"), int),
+            on_road_volume=_number(data.get("OnRoadVolume"), int), yesterday_volume=_number(data.get("YesterdayVolume"), int),
+            last_price=_number(data.get("LastPrice")), profit_rate=_number(data.get("ProfitRate")),
+            future_trade_type=data.get("FutureTradeType"), expire_date=str(data.get("ExpireDate") or ""),
+        )
+
+
+@dataclass
+class Assets:
+    total: float = 0.0
+    available: float = 0.0
+
+
+@dataclass
+class Tick:
+    last_price: float = 0.0
+    last_close: float = 0.0
+    raw: dict[str, Any] = field(default_factory=dict)
+
+
+@dataclass
+class HistoryDataRequest:
+    length: int = 10
+    period: str = ""
+    field: str = ""
+    dividend_type: int = 0
+    skip_paused: bool = True
+
+
+@dataclass
+class MarketDataRequest:
+    fields: list[str] = field(default_factory=list)
+    stocks: list[str] = field(default_factory=list)
+    start_time: str = ""
+    end_time: str = ""
+    period: str = ""
+    dividend_type: str = ""
+    count: int = 0
+
+
+@dataclass
+class FinancialDataRequest:
+    tabname: str = ""; colname: str = ""; market: str = ""; code: str = ""
+    report_type: str = ""; barpos: int = 0
+    field_list: list[str] = field(default_factory=list); stock_list: list[str] = field(default_factory=list)
+    start_date: str = ""; end_date: str = ""
+
+
+@dataclass
+class FactorDataRequest:
+    field_list: list[str] = field(default_factory=list); stock_list: list[str] = field(default_factory=list)
+    stock_code: str = ""; start_date: str = ""; end_date: str = ""
+
+
+@dataclass
+class BSMPriceRequest:
+    option_type: str; object_prices: Any; strike_price: float; risk_free: float; sigma: float; days: int; dividend: float
+
+
+@dataclass
+class BSMIVRequest:
+    option_type: str; object_prices: float; strike_price: float; option_price: float; risk_free: float; days: int; dividend: float
+
+
+@dataclass
+class LocalDataRequest:
+    stock_code: str; start_time: str = ""; end_time: str = ""; period: str = ""; divid_type: str = ""; count: int = 0