mirror of
https://ghfast.top/https://github.com/aeroxw/easy-tdx.git
synced 2026-09-12 13:24:15 +08:00
GitHubandClaude Opus 4.8
4dfd18050e
- pyproject.toml: add mypy overrides for pandas/tabulate/matplotlib stubs, disable strict checking for vendored MyTT library - config.py: use cast() for dict[str, Any] .get() returns - beichi.py: widen _calc_bi_force param to BI | XD, import XD - backtest/cli.py: split combo/single strategy into separate typed variables - backtest/combo.py: add bool_array() helper for numpy return types - chanlun/analyser.py: type ignore for pandas row access, fix dict type arg - unified.py: change fields param from object to Any - ex/mac_client.py: add type args to list literals - cli/cmd_offline.py: wrap int market as Market enum before API call - cli/cmd_chanlun.py: fix dict type arg - offline/write_*.py: explicit int() cast for struct.unpack returns - MyTT.py: fix line-too-long comments, UP038 isinstance syntax - tests: fix E712 (==False → ~mask), E741 (noqa), F841, import sorting - ruff format applied across codebase Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
"""获取证券列表命令（每页最多1000条，按 start 分页）。
|
||
|
||
修复 pytdx Bug #2：GBK 解码使用 errors='replace'，截断多字节序列不再崩溃。
|
||
修复 pytdx Bug #3：pre_close 保持使用通达信自定义浮点解码。
|
||
"""
|
||
|
||
import struct
|
||
|
||
from .._binary import slice_bytes, unpack_from
|
||
from ..codec.volume import _decode_volume
|
||
from ..models.enums import Market
|
||
from ..models.security import SecurityInfo
|
||
from .base import BaseCommand
|
||
|
||
_RECORD_SIZE = 29
|
||
|
||
|
||
class GetSecurityListCmd(BaseCommand[list[SecurityInfo]]):
|
||
"""获取指定市场从 start 开始的证券列表。"""
|
||
|
||
def __init__(self, market: Market, start: int) -> None:
|
||
self.market = market
|
||
self.start = start
|
||
|
||
def build_request(self) -> bytes:
|
||
# Header (12 bytes) + Payload (6 bytes) = 18 bytes
|
||
# Payload: Market(H), Start(H), Unknown(H)=0
|
||
header = bytes.fromhex("0c0118640101060006005004".replace(" ", ""))
|
||
return header + struct.pack("<HHH", int(self.market), self.start, 0)
|
||
|
||
def parse_response(self, body: bytes) -> list[SecurityInfo]:
|
||
(num,) = unpack_from("<H", body, 0, "security_list header")
|
||
pos = 2
|
||
results: list[SecurityInfo] = []
|
||
|
||
for _ in range(num):
|
||
raw = slice_bytes(body, pos, _RECORD_SIZE, "security_list record")
|
||
(
|
||
code_bytes,
|
||
volunit,
|
||
name_bytes,
|
||
_unknown1, # 4字节，排序/分组字段（非用户可见数据）
|
||
decimal_point,
|
||
pre_close_raw,
|
||
_unknown2, # 4字节，私有时间戳（非用户可见数据）
|
||
) = struct.unpack("<6sH8s4sBI4s", raw)
|
||
|
||
code = code_bytes.decode("utf-8", errors="replace").rstrip("\x00")
|
||
# Bug #2 修复：errors='replace' 避免截断 GBK 多字节序列时崩溃
|
||
name = name_bytes.decode("gbk", errors="replace").rstrip("\x00")
|
||
|
||
# pre_close_raw 与协议里的成交量/股本字段一样，使用通达信自定义浮点编码。
|
||
pre_close = _decode_volume(pre_close_raw)
|
||
|
||
results.append(
|
||
SecurityInfo(
|
||
market=self.market,
|
||
code=code,
|
||
name=name,
|
||
volunit=volunit,
|
||
decimal_point=decimal_point,
|
||
pre_close=pre_close,
|
||
_raw=raw,
|
||
)
|
||
)
|
||
pos += _RECORD_SIZE
|
||
|
||
return results
|