mirror of
https://ghfast.top/https://github.com/aeroxw/easy-tdx.git
synced 2026-09-12 22:44:17 +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>
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
"""日期时间解码（通达信 TCP 两种格式）。
|
|
|
|
分钟级（category < 4 或 == 7/8）：4 字节 = 2 字节压缩日期 + 2 字节分钟数
|
|
zipday: year=(>>11)+2004, month=(% 2048)//100, day=(% 2048)%100
|
|
tminutes: hour=//60, minute=%60
|
|
|
|
日线及以上（其余 category）：4 字节 YYYYMMDD 整数
|
|
"""
|
|
|
|
from .._binary import unpack_from
|
|
|
|
|
|
def get_datetime_minute(data: bytes | bytearray, pos: int) -> tuple[int, int, int, int, int, int]:
|
|
"""解析分钟级时间戳（4 字节）。
|
|
|
|
Returns:
|
|
(year, month, day, hour, minute, new_pos)
|
|
"""
|
|
zipday, tminutes = unpack_from("<HH", data, pos, "minute datetime")
|
|
year = (zipday >> 11) + 2004
|
|
month = (zipday % 2048) // 100
|
|
day = (zipday % 2048) % 100
|
|
hour = tminutes // 60
|
|
minute = tminutes % 60
|
|
return year, month, day, hour, minute, pos + 4
|
|
|
|
|
|
def get_datetime_day(data: bytes | bytearray, pos: int) -> tuple[int, int, int, int]:
|
|
"""解析日期（4 字节 YYYYMMDD）。
|
|
|
|
Returns:
|
|
(year, month, day, new_pos)
|
|
"""
|
|
(zipday,) = unpack_from("<I", data, pos, "day datetime")
|
|
year = zipday // 10000
|
|
month = (zipday % 10000) // 100
|
|
day = zipday % 100
|
|
return year, month, day, pos + 4
|
|
|
|
|
|
def get_datetime(
|
|
category: int, data: bytes | bytearray, pos: int
|
|
) -> tuple[int, int, int, int, int, int]:
|
|
"""根据 KlineCategory 选择解析格式。
|
|
|
|
Returns:
|
|
(year, month, day, hour, minute, new_pos)
|
|
日线及以上时 hour=15, minute=0（收盘时间，与 pytdx 保持一致）
|
|
"""
|
|
if category < 4 or category in (7, 8):
|
|
return get_datetime_minute(data, pos)
|
|
else:
|
|
year, month, day, new_pos = get_datetime_day(data, pos)
|
|
return year, month, day, 15, 0, new_pos
|
|
|
|
|
|
def get_time(data: bytes | bytearray, pos: int) -> tuple[int, int, int]:
|
|
"""解析 2 字节时间（分钟数）。
|
|
|
|
Returns:
|
|
(hour, minute, new_pos)
|
|
"""
|
|
(tminutes,) = unpack_from("<H", data, pos, "trade time")
|
|
return tminutes // 60, tminutes % 60, pos + 2
|