mirror of
https://ghfast.top/https://github.com/aeroxw/easy-tdx.git
synced 2026-09-12 16:54:17 +08:00
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
"""历史资金流向命令 (Category 22)。"""
|
|
|
|
import struct
|
|
|
|
from ..codec.volume import _decode_volume
|
|
from ..models.enums import Market
|
|
from ..models.stats import HistoricalFundFlow
|
|
from .base import BaseCommand
|
|
|
|
|
|
class GetHistoryFundFlowCmd(BaseCommand[list[HistoricalFundFlow]]):
|
|
"""获取历史日线资金流向序列。"""
|
|
|
|
def __init__(self, market: Market, code: str, start: int, count: int) -> None:
|
|
self.market = market
|
|
self.code = code.encode("utf-8")
|
|
self.start = start
|
|
self.count = count
|
|
|
|
def build_request(self) -> bytes:
|
|
# Header (12 bytes) + Payload (28 bytes) = 40 bytes
|
|
return struct.pack(
|
|
"<HIHHHH6sHHHHIIH",
|
|
0x010C,
|
|
0x01016408,
|
|
0x001C,
|
|
0x001C,
|
|
0x052D,
|
|
int(self.market),
|
|
self.code,
|
|
22,
|
|
1,
|
|
self.start,
|
|
self.count,
|
|
0,
|
|
0,
|
|
0,
|
|
)
|
|
|
|
def parse_response(self, body: bytes) -> list[HistoricalFundFlow]:
|
|
# 响应格式：9字节头 + 2字节数量 + 每条记录 36 字节
|
|
if len(body) < 11:
|
|
return []
|
|
|
|
(num,) = struct.unpack("<H", body[9:11])
|
|
pos = 11
|
|
results = []
|
|
|
|
for _ in range(num):
|
|
if len(body) < pos + 36:
|
|
break
|
|
|
|
# 记录格式：4字节日期 + 8个4字节自定义浮点金额
|
|
# [0]日期, [1..4]流入(超/大/中/小), [5..8]流出(超/大/中/小)
|
|
raw_data = struct.unpack("<IIIIIIIII", body[pos:pos+36])
|
|
|
|
raw_date = raw_data[0]
|
|
year = raw_date // 10000
|
|
month = (raw_date // 100) % 100
|
|
day = raw_date % 100
|
|
|
|
results.append(HistoricalFundFlow(
|
|
year=year, month=month, day=day,
|
|
super_in=_decode_volume(raw_data[1]),
|
|
large_in=_decode_volume(raw_data[2]),
|
|
medium_in=_decode_volume(raw_data[3]),
|
|
small_in=_decode_volume(raw_data[4]),
|
|
super_out=_decode_volume(raw_data[5]),
|
|
large_out=_decode_volume(raw_data[6]),
|
|
medium_out=_decode_volume(raw_data[7]),
|
|
small_out=_decode_volume(raw_data[8]),
|
|
))
|
|
pos += 36
|
|
|
|
return results
|