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>
117 lines
3.9 KiB
Python
117 lines
3.9 KiB
Python
"""最新财务数据命令。"""
|
|
|
|
import struct
|
|
|
|
from .._binary import slice_bytes, unpack_from
|
|
from ..exceptions import TdxDecodeError
|
|
from ..models.enums import Market
|
|
from ..models.finance import FinanceInfo
|
|
from .base import BaseCommand
|
|
|
|
# 财务字段 struct 格式：1f + 2H + 2I + 30f
|
|
_FIN_FMT = "<fHHII" + "f" * 30
|
|
_FIN_SIZE = struct.calcsize(_FIN_FMT)
|
|
|
|
|
|
class GetFinanceInfoCmd(BaseCommand[FinanceInfo]):
|
|
"""获取单只股票最新财务数据。"""
|
|
|
|
def __init__(self, market: Market, code: str) -> None:
|
|
self.market = market
|
|
self.code = code.encode("utf-8")
|
|
|
|
def build_request(self) -> bytes:
|
|
header = bytes.fromhex("0c1f18760001 0b000b001000 0100".replace(" ", ""))
|
|
return header + struct.pack("<B6s", int(self.market), self.code)
|
|
|
|
def parse_response(self, body: bytes) -> FinanceInfo:
|
|
pos = 2 # 跳过前2字节（记录数）
|
|
market_b, code_b = unpack_from("<B6s", body, pos, "finance_info header")
|
|
pos += 7
|
|
|
|
fields = struct.unpack(_FIN_FMT, slice_bytes(body, pos, _FIN_SIZE, "finance_info body"))
|
|
(
|
|
liutong_guben,
|
|
province,
|
|
industry,
|
|
updated_date,
|
|
ipo_date,
|
|
zong_guben,
|
|
guojia_gu,
|
|
faqiren_faren_gu,
|
|
faren_gu,
|
|
b_gu,
|
|
h_gu,
|
|
zhigong_gu,
|
|
zong_zichan,
|
|
liudong_zichan,
|
|
guding_zichan,
|
|
wuxing_zichan,
|
|
gudong_renshu,
|
|
liudong_fuzhai,
|
|
changqi_fuzhai,
|
|
ziben_gongjijin,
|
|
jing_zichan,
|
|
zhuying_shouru,
|
|
zhuying_lirun,
|
|
yingshou_zhangkuan,
|
|
yingye_lirun,
|
|
touzi_shouyu,
|
|
jingying_xianjinliu,
|
|
zong_xianjinliu,
|
|
cunhuo,
|
|
lirun_zonghe,
|
|
shuihou_lirun,
|
|
jing_lirun,
|
|
weifen_lirun,
|
|
meigujing_zichan,
|
|
reserve2,
|
|
) = fields
|
|
|
|
_SCALE = 10000.0 # 财务数据单位：万元/万股
|
|
try:
|
|
market = Market(market_b)
|
|
except ValueError as e:
|
|
raise TdxDecodeError(f"finance_info 非法 market 值: {market_b}") from e
|
|
|
|
return FinanceInfo(
|
|
market=market,
|
|
code=code_b.decode("utf-8").rstrip("\x00"),
|
|
liutong_guben=liutong_guben * _SCALE,
|
|
zong_guben=zong_guben * _SCALE,
|
|
guojia_gu=guojia_gu * _SCALE,
|
|
faqiren_faren_gu=faqiren_faren_gu * _SCALE,
|
|
faren_gu=faren_gu * _SCALE,
|
|
b_gu=b_gu * _SCALE,
|
|
h_gu=h_gu * _SCALE,
|
|
zhigong_gu=zhigong_gu * _SCALE,
|
|
province=province,
|
|
industry=industry,
|
|
updated_date=updated_date,
|
|
ipo_date=ipo_date,
|
|
gudong_renshu=gudong_renshu,
|
|
zong_zichan=zong_zichan * _SCALE,
|
|
liudong_zichan=liudong_zichan * _SCALE,
|
|
guding_zichan=guding_zichan * _SCALE,
|
|
wuxing_zichan=wuxing_zichan * _SCALE,
|
|
liudong_fuzhai=liudong_fuzhai * _SCALE,
|
|
changqi_fuzhai=changqi_fuzhai * _SCALE,
|
|
ziben_gongjijin=ziben_gongjijin * _SCALE,
|
|
jing_zichan=jing_zichan * _SCALE,
|
|
zhuying_shouru=zhuying_shouru * _SCALE,
|
|
zhuying_lirun=zhuying_lirun * _SCALE,
|
|
yingshou_zhangkuan=yingshou_zhangkuan * _SCALE,
|
|
yingye_lirun=yingye_lirun * _SCALE,
|
|
touzi_shouyu=touzi_shouyu * _SCALE,
|
|
jingying_xianjinliu=jingying_xianjinliu * _SCALE,
|
|
zong_xianjinliu=zong_xianjinliu * _SCALE,
|
|
cunhuo=cunhuo * _SCALE,
|
|
lirun_zonghe=lirun_zonghe * _SCALE,
|
|
shuihou_lirun=shuihou_lirun * _SCALE,
|
|
jing_lirun=jing_lirun * _SCALE,
|
|
weifen_lirun=weifen_lirun * _SCALE,
|
|
meigujing_zichan=meigujing_zichan,
|
|
reserve2=reserve2,
|
|
_raw=body[pos : pos + _FIN_SIZE],
|
|
)
|