107 lines
3.3 KiB
Python
107 lines
3.3 KiB
Python
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
|