pytdxdata 0.1.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (51) hide show
  1. pytdxdata-0.1.0/LICENSE +21 -0
  2. pytdxdata-0.1.0/PKG-INFO +295 -0
  3. pytdxdata-0.1.0/README.md +265 -0
  4. pytdxdata-0.1.0/pyproject.toml +54 -0
  5. pytdxdata-0.1.0/src/pytdxdata/__init__.py +2 -0
  6. pytdxdata-0.1.0/src/pytdxdata/_binary.py +20 -0
  7. pytdxdata-0.1.0/src/pytdxdata/api.py +925 -0
  8. pytdxdata-0.1.0/src/pytdxdata/cache/__init__.py +0 -0
  9. pytdxdata-0.1.0/src/pytdxdata/cache/disk.py +135 -0
  10. pytdxdata-0.1.0/src/pytdxdata/cache/key.py +46 -0
  11. pytdxdata-0.1.0/src/pytdxdata/cache/manager.py +123 -0
  12. pytdxdata-0.1.0/src/pytdxdata/cache/memory.py +59 -0
  13. pytdxdata-0.1.0/src/pytdxdata/cache/ttl.py +50 -0
  14. pytdxdata-0.1.0/src/pytdxdata/cli.py +433 -0
  15. pytdxdata-0.1.0/src/pytdxdata/codec/__init__.py +0 -0
  16. pytdxdata-0.1.0/src/pytdxdata/codec/payload.py +16 -0
  17. pytdxdata-0.1.0/src/pytdxdata/config.py +109 -0
  18. pytdxdata-0.1.0/src/pytdxdata/indicator.py +136 -0
  19. pytdxdata-0.1.0/src/pytdxdata/models/__init__.py +316 -0
  20. pytdxdata-0.1.0/src/pytdxdata/pool/__init__.py +0 -0
  21. pytdxdata-0.1.0/src/pytdxdata/pool/connection.py +105 -0
  22. pytdxdata-0.1.0/src/pytdxdata/pool/ex_connection.py +103 -0
  23. pytdxdata-0.1.0/src/pytdxdata/pool/ex_pool.py +89 -0
  24. pytdxdata-0.1.0/src/pytdxdata/pool/health.py +61 -0
  25. pytdxdata-0.1.0/src/pytdxdata/pool/pool.py +328 -0
  26. pytdxdata-0.1.0/src/pytdxdata/protocol/__init__.py +0 -0
  27. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/__init__.py +32 -0
  28. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/base.py +38 -0
  29. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/ex_ext.py +171 -0
  30. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/ex_proto.py +122 -0
  31. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/finance.py +50 -0
  32. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/kline.py +115 -0
  33. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/mac.py +80 -0
  34. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/mac_ext.py +502 -0
  35. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/minute.py +54 -0
  36. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/quotes.py +118 -0
  37. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/security_list.py +60 -0
  38. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/setup.py +9 -0
  39. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/std_ext.py +195 -0
  40. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/transaction.py +126 -0
  41. pytdxdata-0.1.0/src/pytdxdata/protocol/commands/xdxr.py +65 -0
  42. pytdxdata-0.1.0/src/pytdxdata/protocol/datetime_.py +28 -0
  43. pytdxdata-0.1.0/src/pytdxdata/protocol/frame.py +43 -0
  44. pytdxdata-0.1.0/src/pytdxdata/protocol/mac_frame.py +22 -0
  45. pytdxdata-0.1.0/src/pytdxdata/protocol/price.py +67 -0
  46. pytdxdata-0.1.0/src/pytdxdata/protocol/volume.py +46 -0
  47. pytdxdata-0.1.0/src/pytdxdata/router.py +17 -0
  48. pytdxdata-0.1.0/src/pytdxdata/scheduler/__init__.py +0 -0
  49. pytdxdata-0.1.0/src/pytdxdata/scheduler/paginator.py +128 -0
  50. pytdxdata-0.1.0/src/pytdxdata/server_probe.py +148 -0
  51. pytdxdata-0.1.0/src/pytdxdata/servers.json +423 -0
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2026 vex1023
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
@@ -0,0 +1,295 @@
1
+ Metadata-Version: 2.4
2
+ Name: pytdxdata
3
+ Version: 0.1.0
4
+ Summary: 通达信(TongDaXin)行情数据客户端:动态连接池 + 双层TTL缓存 + 3连接并发分页,纯asyncio零运行时依赖,支持A股/港股/美股/期货/期权
5
+ Keywords: tdx,trading,stock,kline,quote,通达信,A股,港股,美股,期货,asyncio,quant
6
+ Author: vex1023
7
+ Author-email: vex1023 <vex1023@qq.com>
8
+ License-Expression: MIT
9
+ License-File: LICENSE
10
+ Classifier: Development Status :: 4 - Beta
11
+ Classifier: Intended Audience :: Developers
12
+ Classifier: Intended Audience :: Financial and Insurance Industry
13
+ Classifier: Topic :: Office/Business :: Financial :: Investment
14
+ Classifier: Programming Language :: Python :: 3.13
15
+ Classifier: Programming Language :: Python :: 3.12
16
+ Classifier: Operating System :: OS Independent
17
+ Classifier: Framework :: AsyncIO
18
+ Classifier: Typing :: Typed
19
+ Requires-Dist: pytest>=8.0 ; extra == 'dev'
20
+ Requires-Dist: pytest-asyncio>=0.24 ; extra == 'dev'
21
+ Requires-Dist: ruff>=0.6 ; extra == 'dev'
22
+ Requires-Dist: uvloop>=0.21 ; extra == 'uvloop'
23
+ Requires-Python: >=3.13
24
+ Project-URL: Homepage, https://github.com/vex1023/pytdxdata
25
+ Project-URL: Repository, https://github.com/vex1023/pytdxdata
26
+ Project-URL: Documentation, https://github.com/vex1023/pytdxdata#readme
27
+ Provides-Extra: dev
28
+ Provides-Extra: uvloop
29
+ Description-Content-Type: text/markdown
30
+
31
+ # pytdxdata
32
+
33
+ [![PyPI version](https://img.shields.io/pypi/v/pytdxdata)](https://pypi.org/project/pytdxdata/)
34
+ [![Python](https://img.shields.io/pypi/pyversions/pytdxdata)](https://pypi.org/project/pytdxdata/)
35
+ [![License](https://img.shields.io/pypi/l/pytdxdata)](https://github.com/vex1023/pytdxdata/blob/main/LICENSE)
36
+ [![PyPI downloads](https://img.shields.io/pypi/dm/pytdxdata)](https://pypi.org/project/pytdxdata/)
37
+
38
+ 通达信(TongDaXin)行情数据客户端:**动态连接池 + 双层TTL缓存 + 3连接并发分页**(纯 asyncio,零运行时依赖)。
39
+
40
+ **A股(沪/深/北) + 港股 + 美股 + 期货 + 期权**,一套接口全市场。
41
+
42
+ 参照 [easy-tdx](https://github.com/handsomejustin/easy_tdx) 重新实现,解决其三大短板(无连接池、串行分页、单连接单锁),协议层独立实现(不依赖 easy-tdx 包,仅参照其字节级格式)。
43
+
44
+ ## 特性
45
+
46
+ | 特性 | 说明 |
47
+ |------|------|
48
+ | 🧩 动态连接池 | 启动时对服务器**真实K线探测**选优(剔除握手正常但数据返回空的坏节点),按等待队列/空闲时长动态扩缩容,健康分(失败×0.5/成功+0.2/冷却120s)+EWMA延迟选路,跨服务器故障转移 |
49
+ | 🗄️ 双层TTL缓存 | 内存LRU + 磁盘SQLite(WAL),TTL分级:报价5s / 分钟K线60s / 日线当天 / 复权K线1天 / **历史逐笔永久**;single-flight 并发同键去重(只回源一次);缓存"已解析记录"(重复请求0网络0解析) |
50
+ | ⚡ 3连接并发分页 | 单请求拆N页(每页显式start偏移→无状态可并行),3连接并行拉取、按序拼接、失败换连接重试 |
51
+ | 🔀 双通道整合 | 标准通道(经典行情)+ MAC通道(逐笔/复权),按命令自动选路 |
52
+ | 📦 零重型依赖 | 纯 asyncio + stdlib(socket/zlib/struct/sqlite3),返回 `list[dataclass]`,pandas 由调用方按需转换 |
53
+
54
+ ## 安装
55
+
56
+ ```bash
57
+ pip install pytdxdata # 或 uv add pytdxdata
58
+ ```
59
+
60
+ 从源码开发:
61
+
62
+ ```bash
63
+ git clone <repo-url>
64
+ cd pytdxdata
65
+ uv sync # 创建虚拟环境 + 安装CLI
66
+ uv run python -c "import pytdxdata"
67
+ ```
68
+
69
+ 可选加速(uvloop):
70
+
71
+ ```bash
72
+ uv add --optional uvloop uvloop
73
+ ```
74
+
75
+ > 注:数据来自通达信公开行情服务器(无需账号),服务器清单由 `server_probe.py` 导入时自动探测可用性与延时,存于 `servers.json`。
76
+
77
+ ## CLI 命令行工具
78
+
79
+ ```bash
80
+ pip install -e . # 或 uv sync,注册 tdx 命令
81
+
82
+ tdx quotes sz000001 # 行情报价(五档) / hk00700 / usAAPL
83
+ tdx kline sz000001 --period day --count 30 --adjust qfq
84
+ tdx kline cffex:IFL0 --count 30 # 期货K线(hk/us/zz/dl/shf/cffex/gz)
85
+ tdx transactions sz000001 --count 50 # 逐笔(含成交笔数)
86
+ tdx minute sz000001 # 分时
87
+ tdx index 000001 --count 30 # 指数K线
88
+ tdx board list / members 881001 / ranking / change --days 5
89
+ tdx info sz000001 # 个股快照
90
+ tdx list --market sz --count 100 # 证券列表
91
+ tdx auction sz000001 / unusual / xdxr / finance / flow
92
+ tdx server-info / market-stat
93
+ ```
94
+
95
+ ## 快速开始
96
+
97
+ ```python
98
+ import asyncio
99
+ from pytdxdata import TdxData
100
+ from pytdxdata.models import KlinePeriod, Market, Adjust
101
+
102
+ async def main():
103
+ async with TdxData() as td: # 自动建标准池(52台)+MAC池(3台)+缓存
104
+ # 日K线(count>800 自动3连接并发分页)
105
+ bars = await td.get_kline(1, "600000", KlinePeriod.DAY, count=2400)
106
+ # 分钟K线(复权走MAC通道)
107
+ m1 = await td.get_kline(0, "300308", KlinePeriod.MIN_1, count=240)
108
+ # 五档报价(TTL 5s 缓存,重复请求0网络)
109
+ q = await td.get_quotes([(1, "600000"), (0, "000001")])
110
+ # 逐笔成交(MAC通道,含 trade_count=成交笔数,支持历史回溯)
111
+ t = await td.get_transactions(0, "000001", date=20260811)
112
+ total_trades = sum(r.trade_count for r in t) # 当天总成交笔数
113
+ # 证券列表/数量
114
+ n = await td.get_security_count(0)
115
+ lst = await td.get_security_list(1, start=0, count=1000)
116
+ # 分时/除权除息/财务
117
+ minute = await td.get_minute(1, "600000")
118
+ xdxr = await td.get_xdxr(1, "600000")
119
+ fin = await td.get_finance(1, "600000")
120
+
121
+ asyncio.run(main())
122
+ ```
123
+
124
+ ## API 参考
125
+
126
+ 统一入口 `TdxData`(`async with TdxData() as td:` 自动 start/close)。
127
+
128
+ ### 连接管理
129
+
130
+ | 方法 | 说明 |
131
+ |------|------|
132
+ | `start()` / `close()` | 启动(服务器探测+预建连接+缓存)/ 关闭 |
133
+ | `__aenter__` / `__aexit__` | 异步上下文管理器 |
134
+
135
+ ### 行情接口
136
+
137
+ | 方法 | 返回 | 说明 |
138
+ |------|------|------|
139
+ | `get_kline(market, code, period, *, start, count, adjust)` | `list[SecurityBar]` | K线(count>800自动并发分页;adjust≠NONE走MAC复权;88/000/399指数走指数命令) |
140
+ | `get_index_kline(market, code, period, *, start, count)` | `list[SecurityBar]` | 指数K线(多涨跌家数字段已处理) |
141
+ | `get_quotes(stocks)` | `list[SecurityQuote]` | 五档报价(>80只自动切批,缓存键=规范化股票集) |
142
+ | `get_transactions(market, code, *, date, start, count)` | `list[TransactionRecord]` | **逐笔成交(MAC,含 trade_count 成交笔数)**,支持历史回溯 |
143
+ | `get_minute(market, code, *, date)` | `list[MinuteBar]` | 分时(今日/历史) |
144
+ | `get_security_count(market)` | `int` | 市场证券数量 |
145
+ | `get_security_list(market, *, start, count)` | `list[SecurityInfo]` | 证券列表(count=None=全部,先查总数分页) |
146
+ | `get_security_list_all(market)` | `list[SecurityInfo]` | 全市场列表(缓存1天) |
147
+ | `get_xdxr(market, code)` | `list[XdxrRecord]` | 除权除息历史(标准) |
148
+ | `get_finance(market, code)` | `FinanceRecord \| None` | 财务快照(标准) |
149
+
150
+ ### 板块/竞价/异动(MAC)
151
+
152
+ | 方法 | 返回 | 说明 |
153
+ |------|------|------|
154
+ | `get_board_list(board_type, count)` | `list[BoardInfo]` | 板块列表(0=全部 1=行业 2=概念) |
155
+ | `get_board_members(board_symbol, count, sort_type)` | `list[MemberQuote]` | 板块成分股报价(自定义字段) |
156
+ | `get_board_summary(board_symbol)` | `dict` | 板块汇总(成分数/成交额/主力净流入/涨跌家数) |
157
+ | `get_board_ranking(board_type, top_n)` | `list[dict]` | 板块涨跌幅排行 |
158
+ | `get_board_change_ranking(board_type, days, top_n)` | `list[dict]` | 板块N日涨跌幅排行(板块指数K线) |
159
+ | `get_belong_board(market, code)` | `list[BelongBoard]` | 个股所属板块 |
160
+ | `get_auction(market, code)` | `list[AuctionItem]` | 集合竞价 |
161
+ | `get_unusual(market, start, count)` | `list[UnusualItem]` | 市场异动 |
162
+ | `get_capital_flow(market, code)` | `CapitalFlow` | 资金流向 |
163
+
164
+ ### 快照/服务器/扩展(MAC)
165
+
166
+ | 方法 | 返回 | 说明 |
167
+ |------|------|------|
168
+ | `get_symbol_info(market, code)` | `SymbolSnapshot` | 个股特征快照 |
169
+ | `get_stock_quotes(stocks, bits)` | `list[MemberQuote]` | MAC自定义字段报价 |
170
+ | `get_stock_quotes_list(category, count)` | `list[MemberQuote]` | 市场分类报价列表(0沪A/6全A/8科创/14创业) |
171
+ | `get_stock_kline_with_indicators(market, code, indicators, ...)` | `dict` | K线+指标(MACD/KDJ/RSI/BOLL/MA/EMA) |
172
+ | `get_goods_list(market, start, count)` | `list[GoodsItem]` | 扩展市场商品(期货/期权) |
173
+ | `get_server_info()` | `ServerInfo` | 服务器交易时段 |
174
+ | `get_kline_offset(offset, count)` | `tuple[int,int]` | K线偏移 |
175
+ | `get_chart_sampling(market, code)` | `list[float]` | 分时采样(扩展市场命令,A股不适用) |
176
+
177
+ ### 文件/信息(标准+MAC)
178
+
179
+ | 方法 | 返回 | 说明 |
180
+ |------|------|------|
181
+ | `get_company_info_category/content(...)` | `list` / `str` | F10公司信息(标准) |
182
+ | `get_block_info(filename)` | `bytes` | 板块文件下载(标准) |
183
+ | `get_report_file(filename)` | `bytes` | 报告/财务文件下载(标准) |
184
+ | `get_financial_file_list()` | `list[str]` | 财务文件列表(tdxfin/gpcw.txt) |
185
+ | `get_financial_records(filename)` | `bytes` | 财务文件内容(gpcw*.zip) |
186
+ | `get_file_meta(filename)` / `download_file(filename)` | `FileMeta` / `bytes` | 远程文件(MAC) |
187
+ | `get_price_limits(market, code, name, pre_close, listed_days)` | `tuple` | 涨跌停价(本地计算) |
188
+ | `get_market_stat()` | `MarketStat` | 全市场统计(涨跌家数/总额/市值/涨跌停数) |
189
+
190
+ ### 数据模型(dataclass)
191
+
192
+ | 模型 | 关键字段 |
193
+ |------|---------|
194
+ | `SecurityBar` | market/code/open/high/low/close/vol/amount/year/month/day/hour/minute + `datetime`属性 |
195
+ | `SecurityQuote` | price/pre_close/OHLC/vol/amount/s_vol/b_vol/`bid[5]`/`ask[5]`/server_time/trading_status |
196
+ | `TransactionRecord` | time/price/vol/`trade_count`/bs_flag |
197
+ | `SecurityInfo` | code/name(GBK)/volunit/decimal_point/pre_close |
198
+ | `XdxrRecord` | category/分红送转/缩股/行权等(除权除息) |
199
+ | `MinuteBar` | price/vol |
200
+ | `FinanceRecord` | 股本/资产/营收/利润(万元/万股) |
201
+
202
+ ### 枚举
203
+
204
+ - `Market`:SZ=0 / SH=1 / **BJ=2**(北交所)
205
+ - `KlinePeriod`:MIN_5=0 / MIN_15=1 / MIN_30=2 / MIN_60=3 / **DAY=4** / WEEK=5 / MONTH=6 / MIN_1=7 / MIN_3=8 / YEAR=9 / SEASON=10(**与通达信协议字节一致**)
206
+ - `Adjust`:NONE / QFQ(前复权)/ HFQ(后复权)
207
+
208
+ ## 通道路由
209
+
210
+ | 命令 | 默认通道 | 说明 |
211
+ |------|---------|------|
212
+ | K线(复权) | **MAC** | 支持QFQ/HFQ |
213
+ | K线(不复权)/指数 | 标准 | 价格小数位规则已验证 |
214
+ | 报价 | 标准 | 股票2位/指数ETF转债3位 |
215
+ | **逐笔成交** | **MAC** | 含trade_count,支持历史日期 |
216
+ | 列表/数量/分时/除权/财务 | 标准 | 仅标准通道提供 |
217
+
218
+ ## 架构
219
+
220
+ ```
221
+ src/pytdxdata/
222
+ ├── api.py # 统一接口 TdxData(对外唯一入口)
223
+ ├── router.py # 标准/MAC 通道选路
224
+ ├── config.py # 服务器清单加载(servers.json + 环境变量覆盖)
225
+ ├── protocol/ # 独立协议层(帧/变长整数/差分还原/自定义浮点/命令)
226
+ │ └── commands/ # kline/quotes/list/transaction/minute/xdxr/finance/mac
227
+ ├── pool/ # 动态连接池(connection/health/pool)
228
+ ├── cache/ # 双层TTL缓存(key/ttl/memory/disk/manager)
229
+ ├── scheduler/ # 并发分页调度器(Paginator,3连接并行)
230
+ ├── models/ # dataclass 模型 + 枚举
231
+ └── codec/ # 缓存载荷序列化(pickle+zlib)
232
+ ```
233
+
234
+ ## 扩展市场(EX通道:港股/美股/期货/期权)
235
+
236
+ `get_kline` 的 market 参数接受 `ExMarket` 枚举(港股/美股/期货/期权等52个市场):
237
+
238
+ ```python
239
+ from pytdxdata.models import ExMarket, KlinePeriod
240
+
241
+ # 港股(HK_MAIN_BOARD=31,代码5位)
242
+ bars = await td.get_kline(ExMarket.HK_MAIN_BOARD, '00700', KlinePeriod.DAY, count=700)
243
+ # 美股(US_STOCK=74)
244
+ bars = await td.get_kline(ExMarket.US_STOCK, 'AAPL', KlinePeriod.DAY, count=700)
245
+ # 期货(CFFEX=47 / 郑28 / 大29 / 上30,合约如 IFL0/rb2610)
246
+ bars = await td.get_kline(ExMarket.CFFEX_FUTURES, 'IFL0', KlinePeriod.DAY, count=700)
247
+ # 股票期权(沪8/深9,合约代码8位)
248
+ bars = await td.get_kline(ExMarket.SH_STOCK_OPTION, '10010971', KlinePeriod.DAY, count=700)
249
+ ```
250
+
251
+ - 端口 7727,统一走 **pytdx EX 协议**(0x23FF K线 / 0x23FA 报价 / 0x23F4 市场列表)
252
+ - 4台EX服务器:MAC EX 2台(116.205.135.205 / 121.37.232.167,0x2454 login)+ pytdx EX 2台(通达信扩展市场 116.205.143.214 / 国泰君安 103.221.142.82,0x2454 setup)
253
+ - EX 辅助接口:get_goods_list / get_goods_count / get_goods_quotes / get_goods_tick_chart / get_goods_transaction 等
254
+ - 实测:腾讯00700=466.6 / AAPL=332.8 / IFL0=4683.2 / 沪期权10010971=0.192
255
+ - 美股/港股期权:通达信协议无数据(美股期权可用 CBOE 公开接口,见 quantdata/scripts/import/cboe_options.py)
256
+
257
+ ## 与 easy-tdx 对比
258
+
259
+ | 维度 | easy-tdx | pytdxdata |
260
+ |------|----------|----------|
261
+ | 并发分页 | ❌ 串行 while 循环 | ✅ 3连接并行 |
262
+ | 连接池 | ❌ 单连接单锁 | ✅ 动态池+故障转移 |
263
+ | 缓存 | 仅证券列表(1天) | ✅ 双层TTL+single-flight |
264
+ | 数据正确性 | 基准 | ✅ K线6字段/报价7字段逐字段一致(实测) |
265
+ | 2400根日K耗时 | 0.31s | **0.16s(1.97x)** |
266
+ | 缓存命中重取 | — | 0.016s(0网络) |
267
+ | 接口覆盖 | 基准 | **36/36 全覆盖** |
268
+ | 依赖 | pandas等 | 零运行时依赖 |
269
+
270
+ 实测对比报告:`examples/compare_report.md`
271
+
272
+ ## 已知限制
273
+
274
+ - 标准通道K线 `count<100` 部分服务器返回空 → 分页器已强制 `min_page=100` 规避
275
+ - 标准当日逐笔的笔数字段协议中有但 easy-tdx 模型丢弃(pytdxdata 已显式解析);历史逐笔无笔数字段
276
+ - 报价价格小数位按品种推断(股票2位/指数ETF转债3位),同代码不同市场含义不同(SH 000001=指数 vs SZ 000001=股票)
277
+ - 行情数据来自通达信公开服务器,**15分钟延时**;需实时数据请使用券商行情
278
+ - 港股/美股/期货/期权数据走通达信 EX 通道(7727),部分市场(美股/港股期权)协议内无数据
279
+ - 服务器可用性随时间变化,可用 `python -m pytdxdata.server_probe` 重新探测
280
+
281
+ ## 免责声明
282
+
283
+ 本包仅用于**技术研究与教育目的**。行情数据来自通达信公开接口,不保证实时性与准确性;作者不对任何投资决策负责。使用前请遵守通达信服务条款及相关法律法规。
284
+
285
+ ## 测试
286
+
287
+ ```bash
288
+ uv run pytest tests/ # 48个测试:协议/池/缓存/分页/API
289
+ ```
290
+
291
+ ## 目录
292
+
293
+ - `examples/quickstart.py` — 快速示例
294
+ - `examples/tdx_update_minute.py` — 全市场分钟K线更新(含北交所,45s拉完5,539只)
295
+ - `examples/compare_easy_tdx.py` — 与 easy-tdx 对比脚本
@@ -0,0 +1,265 @@
1
+ # pytdxdata
2
+
3
+ [![PyPI version](https://img.shields.io/pypi/v/pytdxdata)](https://pypi.org/project/pytdxdata/)
4
+ [![Python](https://img.shields.io/pypi/pyversions/pytdxdata)](https://pypi.org/project/pytdxdata/)
5
+ [![License](https://img.shields.io/pypi/l/pytdxdata)](https://github.com/vex1023/pytdxdata/blob/main/LICENSE)
6
+ [![PyPI downloads](https://img.shields.io/pypi/dm/pytdxdata)](https://pypi.org/project/pytdxdata/)
7
+
8
+ 通达信(TongDaXin)行情数据客户端:**动态连接池 + 双层TTL缓存 + 3连接并发分页**(纯 asyncio,零运行时依赖)。
9
+
10
+ **A股(沪/深/北) + 港股 + 美股 + 期货 + 期权**,一套接口全市场。
11
+
12
+ 参照 [easy-tdx](https://github.com/handsomejustin/easy_tdx) 重新实现,解决其三大短板(无连接池、串行分页、单连接单锁),协议层独立实现(不依赖 easy-tdx 包,仅参照其字节级格式)。
13
+
14
+ ## 特性
15
+
16
+ | 特性 | 说明 |
17
+ |------|------|
18
+ | 🧩 动态连接池 | 启动时对服务器**真实K线探测**选优(剔除握手正常但数据返回空的坏节点),按等待队列/空闲时长动态扩缩容,健康分(失败×0.5/成功+0.2/冷却120s)+EWMA延迟选路,跨服务器故障转移 |
19
+ | 🗄️ 双层TTL缓存 | 内存LRU + 磁盘SQLite(WAL),TTL分级:报价5s / 分钟K线60s / 日线当天 / 复权K线1天 / **历史逐笔永久**;single-flight 并发同键去重(只回源一次);缓存"已解析记录"(重复请求0网络0解析) |
20
+ | ⚡ 3连接并发分页 | 单请求拆N页(每页显式start偏移→无状态可并行),3连接并行拉取、按序拼接、失败换连接重试 |
21
+ | 🔀 双通道整合 | 标准通道(经典行情)+ MAC通道(逐笔/复权),按命令自动选路 |
22
+ | 📦 零重型依赖 | 纯 asyncio + stdlib(socket/zlib/struct/sqlite3),返回 `list[dataclass]`,pandas 由调用方按需转换 |
23
+
24
+ ## 安装
25
+
26
+ ```bash
27
+ pip install pytdxdata # 或 uv add pytdxdata
28
+ ```
29
+
30
+ 从源码开发:
31
+
32
+ ```bash
33
+ git clone <repo-url>
34
+ cd pytdxdata
35
+ uv sync # 创建虚拟环境 + 安装CLI
36
+ uv run python -c "import pytdxdata"
37
+ ```
38
+
39
+ 可选加速(uvloop):
40
+
41
+ ```bash
42
+ uv add --optional uvloop uvloop
43
+ ```
44
+
45
+ > 注:数据来自通达信公开行情服务器(无需账号),服务器清单由 `server_probe.py` 导入时自动探测可用性与延时,存于 `servers.json`。
46
+
47
+ ## CLI 命令行工具
48
+
49
+ ```bash
50
+ pip install -e . # 或 uv sync,注册 tdx 命令
51
+
52
+ tdx quotes sz000001 # 行情报价(五档) / hk00700 / usAAPL
53
+ tdx kline sz000001 --period day --count 30 --adjust qfq
54
+ tdx kline cffex:IFL0 --count 30 # 期货K线(hk/us/zz/dl/shf/cffex/gz)
55
+ tdx transactions sz000001 --count 50 # 逐笔(含成交笔数)
56
+ tdx minute sz000001 # 分时
57
+ tdx index 000001 --count 30 # 指数K线
58
+ tdx board list / members 881001 / ranking / change --days 5
59
+ tdx info sz000001 # 个股快照
60
+ tdx list --market sz --count 100 # 证券列表
61
+ tdx auction sz000001 / unusual / xdxr / finance / flow
62
+ tdx server-info / market-stat
63
+ ```
64
+
65
+ ## 快速开始
66
+
67
+ ```python
68
+ import asyncio
69
+ from pytdxdata import TdxData
70
+ from pytdxdata.models import KlinePeriod, Market, Adjust
71
+
72
+ async def main():
73
+ async with TdxData() as td: # 自动建标准池(52台)+MAC池(3台)+缓存
74
+ # 日K线(count>800 自动3连接并发分页)
75
+ bars = await td.get_kline(1, "600000", KlinePeriod.DAY, count=2400)
76
+ # 分钟K线(复权走MAC通道)
77
+ m1 = await td.get_kline(0, "300308", KlinePeriod.MIN_1, count=240)
78
+ # 五档报价(TTL 5s 缓存,重复请求0网络)
79
+ q = await td.get_quotes([(1, "600000"), (0, "000001")])
80
+ # 逐笔成交(MAC通道,含 trade_count=成交笔数,支持历史回溯)
81
+ t = await td.get_transactions(0, "000001", date=20260811)
82
+ total_trades = sum(r.trade_count for r in t) # 当天总成交笔数
83
+ # 证券列表/数量
84
+ n = await td.get_security_count(0)
85
+ lst = await td.get_security_list(1, start=0, count=1000)
86
+ # 分时/除权除息/财务
87
+ minute = await td.get_minute(1, "600000")
88
+ xdxr = await td.get_xdxr(1, "600000")
89
+ fin = await td.get_finance(1, "600000")
90
+
91
+ asyncio.run(main())
92
+ ```
93
+
94
+ ## API 参考
95
+
96
+ 统一入口 `TdxData`(`async with TdxData() as td:` 自动 start/close)。
97
+
98
+ ### 连接管理
99
+
100
+ | 方法 | 说明 |
101
+ |------|------|
102
+ | `start()` / `close()` | 启动(服务器探测+预建连接+缓存)/ 关闭 |
103
+ | `__aenter__` / `__aexit__` | 异步上下文管理器 |
104
+
105
+ ### 行情接口
106
+
107
+ | 方法 | 返回 | 说明 |
108
+ |------|------|------|
109
+ | `get_kline(market, code, period, *, start, count, adjust)` | `list[SecurityBar]` | K线(count>800自动并发分页;adjust≠NONE走MAC复权;88/000/399指数走指数命令) |
110
+ | `get_index_kline(market, code, period, *, start, count)` | `list[SecurityBar]` | 指数K线(多涨跌家数字段已处理) |
111
+ | `get_quotes(stocks)` | `list[SecurityQuote]` | 五档报价(>80只自动切批,缓存键=规范化股票集) |
112
+ | `get_transactions(market, code, *, date, start, count)` | `list[TransactionRecord]` | **逐笔成交(MAC,含 trade_count 成交笔数)**,支持历史回溯 |
113
+ | `get_minute(market, code, *, date)` | `list[MinuteBar]` | 分时(今日/历史) |
114
+ | `get_security_count(market)` | `int` | 市场证券数量 |
115
+ | `get_security_list(market, *, start, count)` | `list[SecurityInfo]` | 证券列表(count=None=全部,先查总数分页) |
116
+ | `get_security_list_all(market)` | `list[SecurityInfo]` | 全市场列表(缓存1天) |
117
+ | `get_xdxr(market, code)` | `list[XdxrRecord]` | 除权除息历史(标准) |
118
+ | `get_finance(market, code)` | `FinanceRecord \| None` | 财务快照(标准) |
119
+
120
+ ### 板块/竞价/异动(MAC)
121
+
122
+ | 方法 | 返回 | 说明 |
123
+ |------|------|------|
124
+ | `get_board_list(board_type, count)` | `list[BoardInfo]` | 板块列表(0=全部 1=行业 2=概念) |
125
+ | `get_board_members(board_symbol, count, sort_type)` | `list[MemberQuote]` | 板块成分股报价(自定义字段) |
126
+ | `get_board_summary(board_symbol)` | `dict` | 板块汇总(成分数/成交额/主力净流入/涨跌家数) |
127
+ | `get_board_ranking(board_type, top_n)` | `list[dict]` | 板块涨跌幅排行 |
128
+ | `get_board_change_ranking(board_type, days, top_n)` | `list[dict]` | 板块N日涨跌幅排行(板块指数K线) |
129
+ | `get_belong_board(market, code)` | `list[BelongBoard]` | 个股所属板块 |
130
+ | `get_auction(market, code)` | `list[AuctionItem]` | 集合竞价 |
131
+ | `get_unusual(market, start, count)` | `list[UnusualItem]` | 市场异动 |
132
+ | `get_capital_flow(market, code)` | `CapitalFlow` | 资金流向 |
133
+
134
+ ### 快照/服务器/扩展(MAC)
135
+
136
+ | 方法 | 返回 | 说明 |
137
+ |------|------|------|
138
+ | `get_symbol_info(market, code)` | `SymbolSnapshot` | 个股特征快照 |
139
+ | `get_stock_quotes(stocks, bits)` | `list[MemberQuote]` | MAC自定义字段报价 |
140
+ | `get_stock_quotes_list(category, count)` | `list[MemberQuote]` | 市场分类报价列表(0沪A/6全A/8科创/14创业) |
141
+ | `get_stock_kline_with_indicators(market, code, indicators, ...)` | `dict` | K线+指标(MACD/KDJ/RSI/BOLL/MA/EMA) |
142
+ | `get_goods_list(market, start, count)` | `list[GoodsItem]` | 扩展市场商品(期货/期权) |
143
+ | `get_server_info()` | `ServerInfo` | 服务器交易时段 |
144
+ | `get_kline_offset(offset, count)` | `tuple[int,int]` | K线偏移 |
145
+ | `get_chart_sampling(market, code)` | `list[float]` | 分时采样(扩展市场命令,A股不适用) |
146
+
147
+ ### 文件/信息(标准+MAC)
148
+
149
+ | 方法 | 返回 | 说明 |
150
+ |------|------|------|
151
+ | `get_company_info_category/content(...)` | `list` / `str` | F10公司信息(标准) |
152
+ | `get_block_info(filename)` | `bytes` | 板块文件下载(标准) |
153
+ | `get_report_file(filename)` | `bytes` | 报告/财务文件下载(标准) |
154
+ | `get_financial_file_list()` | `list[str]` | 财务文件列表(tdxfin/gpcw.txt) |
155
+ | `get_financial_records(filename)` | `bytes` | 财务文件内容(gpcw*.zip) |
156
+ | `get_file_meta(filename)` / `download_file(filename)` | `FileMeta` / `bytes` | 远程文件(MAC) |
157
+ | `get_price_limits(market, code, name, pre_close, listed_days)` | `tuple` | 涨跌停价(本地计算) |
158
+ | `get_market_stat()` | `MarketStat` | 全市场统计(涨跌家数/总额/市值/涨跌停数) |
159
+
160
+ ### 数据模型(dataclass)
161
+
162
+ | 模型 | 关键字段 |
163
+ |------|---------|
164
+ | `SecurityBar` | market/code/open/high/low/close/vol/amount/year/month/day/hour/minute + `datetime`属性 |
165
+ | `SecurityQuote` | price/pre_close/OHLC/vol/amount/s_vol/b_vol/`bid[5]`/`ask[5]`/server_time/trading_status |
166
+ | `TransactionRecord` | time/price/vol/`trade_count`/bs_flag |
167
+ | `SecurityInfo` | code/name(GBK)/volunit/decimal_point/pre_close |
168
+ | `XdxrRecord` | category/分红送转/缩股/行权等(除权除息) |
169
+ | `MinuteBar` | price/vol |
170
+ | `FinanceRecord` | 股本/资产/营收/利润(万元/万股) |
171
+
172
+ ### 枚举
173
+
174
+ - `Market`:SZ=0 / SH=1 / **BJ=2**(北交所)
175
+ - `KlinePeriod`:MIN_5=0 / MIN_15=1 / MIN_30=2 / MIN_60=3 / **DAY=4** / WEEK=5 / MONTH=6 / MIN_1=7 / MIN_3=8 / YEAR=9 / SEASON=10(**与通达信协议字节一致**)
176
+ - `Adjust`:NONE / QFQ(前复权)/ HFQ(后复权)
177
+
178
+ ## 通道路由
179
+
180
+ | 命令 | 默认通道 | 说明 |
181
+ |------|---------|------|
182
+ | K线(复权) | **MAC** | 支持QFQ/HFQ |
183
+ | K线(不复权)/指数 | 标准 | 价格小数位规则已验证 |
184
+ | 报价 | 标准 | 股票2位/指数ETF转债3位 |
185
+ | **逐笔成交** | **MAC** | 含trade_count,支持历史日期 |
186
+ | 列表/数量/分时/除权/财务 | 标准 | 仅标准通道提供 |
187
+
188
+ ## 架构
189
+
190
+ ```
191
+ src/pytdxdata/
192
+ ├── api.py # 统一接口 TdxData(对外唯一入口)
193
+ ├── router.py # 标准/MAC 通道选路
194
+ ├── config.py # 服务器清单加载(servers.json + 环境变量覆盖)
195
+ ├── protocol/ # 独立协议层(帧/变长整数/差分还原/自定义浮点/命令)
196
+ │ └── commands/ # kline/quotes/list/transaction/minute/xdxr/finance/mac
197
+ ├── pool/ # 动态连接池(connection/health/pool)
198
+ ├── cache/ # 双层TTL缓存(key/ttl/memory/disk/manager)
199
+ ├── scheduler/ # 并发分页调度器(Paginator,3连接并行)
200
+ ├── models/ # dataclass 模型 + 枚举
201
+ └── codec/ # 缓存载荷序列化(pickle+zlib)
202
+ ```
203
+
204
+ ## 扩展市场(EX通道:港股/美股/期货/期权)
205
+
206
+ `get_kline` 的 market 参数接受 `ExMarket` 枚举(港股/美股/期货/期权等52个市场):
207
+
208
+ ```python
209
+ from pytdxdata.models import ExMarket, KlinePeriod
210
+
211
+ # 港股(HK_MAIN_BOARD=31,代码5位)
212
+ bars = await td.get_kline(ExMarket.HK_MAIN_BOARD, '00700', KlinePeriod.DAY, count=700)
213
+ # 美股(US_STOCK=74)
214
+ bars = await td.get_kline(ExMarket.US_STOCK, 'AAPL', KlinePeriod.DAY, count=700)
215
+ # 期货(CFFEX=47 / 郑28 / 大29 / 上30,合约如 IFL0/rb2610)
216
+ bars = await td.get_kline(ExMarket.CFFEX_FUTURES, 'IFL0', KlinePeriod.DAY, count=700)
217
+ # 股票期权(沪8/深9,合约代码8位)
218
+ bars = await td.get_kline(ExMarket.SH_STOCK_OPTION, '10010971', KlinePeriod.DAY, count=700)
219
+ ```
220
+
221
+ - 端口 7727,统一走 **pytdx EX 协议**(0x23FF K线 / 0x23FA 报价 / 0x23F4 市场列表)
222
+ - 4台EX服务器:MAC EX 2台(116.205.135.205 / 121.37.232.167,0x2454 login)+ pytdx EX 2台(通达信扩展市场 116.205.143.214 / 国泰君安 103.221.142.82,0x2454 setup)
223
+ - EX 辅助接口:get_goods_list / get_goods_count / get_goods_quotes / get_goods_tick_chart / get_goods_transaction 等
224
+ - 实测:腾讯00700=466.6 / AAPL=332.8 / IFL0=4683.2 / 沪期权10010971=0.192
225
+ - 美股/港股期权:通达信协议无数据(美股期权可用 CBOE 公开接口,见 quantdata/scripts/import/cboe_options.py)
226
+
227
+ ## 与 easy-tdx 对比
228
+
229
+ | 维度 | easy-tdx | pytdxdata |
230
+ |------|----------|----------|
231
+ | 并发分页 | ❌ 串行 while 循环 | ✅ 3连接并行 |
232
+ | 连接池 | ❌ 单连接单锁 | ✅ 动态池+故障转移 |
233
+ | 缓存 | 仅证券列表(1天) | ✅ 双层TTL+single-flight |
234
+ | 数据正确性 | 基准 | ✅ K线6字段/报价7字段逐字段一致(实测) |
235
+ | 2400根日K耗时 | 0.31s | **0.16s(1.97x)** |
236
+ | 缓存命中重取 | — | 0.016s(0网络) |
237
+ | 接口覆盖 | 基准 | **36/36 全覆盖** |
238
+ | 依赖 | pandas等 | 零运行时依赖 |
239
+
240
+ 实测对比报告:`examples/compare_report.md`
241
+
242
+ ## 已知限制
243
+
244
+ - 标准通道K线 `count<100` 部分服务器返回空 → 分页器已强制 `min_page=100` 规避
245
+ - 标准当日逐笔的笔数字段协议中有但 easy-tdx 模型丢弃(pytdxdata 已显式解析);历史逐笔无笔数字段
246
+ - 报价价格小数位按品种推断(股票2位/指数ETF转债3位),同代码不同市场含义不同(SH 000001=指数 vs SZ 000001=股票)
247
+ - 行情数据来自通达信公开服务器,**15分钟延时**;需实时数据请使用券商行情
248
+ - 港股/美股/期货/期权数据走通达信 EX 通道(7727),部分市场(美股/港股期权)协议内无数据
249
+ - 服务器可用性随时间变化,可用 `python -m pytdxdata.server_probe` 重新探测
250
+
251
+ ## 免责声明
252
+
253
+ 本包仅用于**技术研究与教育目的**。行情数据来自通达信公开接口,不保证实时性与准确性;作者不对任何投资决策负责。使用前请遵守通达信服务条款及相关法律法规。
254
+
255
+ ## 测试
256
+
257
+ ```bash
258
+ uv run pytest tests/ # 48个测试:协议/池/缓存/分页/API
259
+ ```
260
+
261
+ ## 目录
262
+
263
+ - `examples/quickstart.py` — 快速示例
264
+ - `examples/tdx_update_minute.py` — 全市场分钟K线更新(含北交所,45s拉完5,539只)
265
+ - `examples/compare_easy_tdx.py` — 与 easy-tdx 对比脚本
@@ -0,0 +1,54 @@
1
+ [project]
2
+ name = "pytdxdata"
3
+ version = "0.1.0"
4
+ description = "通达信(TongDaXin)行情数据客户端:动态连接池 + 双层TTL缓存 + 3连接并发分页,纯asyncio零运行时依赖,支持A股/港股/美股/期货/期权"
5
+ readme = "README.md"
6
+ requires-python = ">=3.13"
7
+ license = "MIT"
8
+ license-files = ["LICENSE"]
9
+ keywords = ["tdx", "trading", "stock", "kline", "quote", "通达信", "A股", "港股", "美股", "期货", "asyncio", "quant"]
10
+ authors = [
11
+ { name = "vex1023", email = "vex1023@qq.com" }
12
+ ]
13
+ classifiers = [
14
+ "Development Status :: 4 - Beta",
15
+ "Intended Audience :: Developers",
16
+ "Intended Audience :: Financial and Insurance Industry",
17
+ "Topic :: Office/Business :: Financial :: Investment",
18
+ "Programming Language :: Python :: 3.13",
19
+ "Programming Language :: Python :: 3.12",
20
+ "Operating System :: OS Independent",
21
+ "Framework :: AsyncIO",
22
+ "Typing :: Typed",
23
+ ]
24
+ dependencies = []
25
+
26
+ [project.urls]
27
+ Homepage = "https://github.com/vex1023/pytdxdata"
28
+ Repository = "https://github.com/vex1023/pytdxdata"
29
+ Documentation = "https://github.com/vex1023/pytdxdata#readme"
30
+
31
+ [project.optional-dependencies]
32
+ uvloop = ["uvloop>=0.21"]
33
+ dev = ["pytest>=8.0", "pytest-asyncio>=0.24", "ruff>=0.6"]
34
+
35
+ [project.scripts]
36
+ tdx = "pytdxdata.cli:main"
37
+
38
+ [build-system]
39
+ requires = ["uv_build>=0.11.1,<0.12.0"]
40
+ build-backend = "uv_build"
41
+
42
+ [tool.ruff]
43
+ line-length = 100
44
+ target-version = "py313"
45
+
46
+ [tool.pytest.ini_options]
47
+ asyncio_mode = "auto"
48
+
49
+ [dependency-groups]
50
+ dev = [
51
+ "pytest>=9.1.1",
52
+ "pytest-asyncio>=1.4.0",
53
+ ]
54
+
@@ -0,0 +1,2 @@
1
+ def main() -> None:
2
+ print("Hello from pytdxdata!")
@@ -0,0 +1,20 @@
1
+ """二进制解包工具"""
2
+ from __future__ import annotations
3
+
4
+ import struct
5
+ from typing import Any
6
+
7
+
8
+ def unpack_from(fmt: str, data: bytes | bytearray | memoryview, pos: int, what: str = "") -> tuple[Any, ...]:
9
+ """从 pos 处按 fmt 解包,越界抛清晰错误。"""
10
+ try:
11
+ return struct.unpack_from(fmt, data, pos)
12
+ except struct.error as e:
13
+ raise ValueError(f"{what} 解包失败 @{pos} fmt={fmt}: {e}") from e
14
+
15
+
16
+ def slice_bytes(data: bytes | bytearray | memoryview, pos: int, length: int, what: str = "") -> bytes:
17
+ end = pos + length
18
+ if end > len(data):
19
+ raise ValueError(f"{what} 截断 @{pos} need {length} have {len(data) - pos}")
20
+ return bytes(data[pos:end])