qdata-quote 0.2.2__tar.gz → 0.3.0__tar.gz

{qdata_quote-0.2.2 → qdata_quote-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qdata-quote
-Version: 0.2.2
+Version: 0.3.0
 Summary: 实时行情采集服务，支持新浪和腾讯数据源
 License-Expression: MIT
 Requires-Python: >=3.10
@@ -25,12 +25,13 @@ pip install qdata-quote
 ## 快速开始
 
 ```python
-from qdata_quote import QuoteService
+from qdata_quote import QuoteService, get_stock_codes
 
-service = QuoteService()
+# 获取全市场股票代码（首次从网络获取约 4 秒，后续读缓存 <1ms）
+codes = get_stock_codes()
 
-# 设置全市场股票代码（必需）
-service.set_stock_codes(["000001", "600000", "600036", "300750"])
+service = QuoteService()
+service.set_stock_codes(codes)
 
 # 同步获取指定股票行情
 df = service.get_real_sync(["000001", "600000"])
@@ -45,6 +46,20 @@ df = asyncio.run(service.get_real(["000001", "600000"]))
 df_all = asyncio.run(service.get_all())
 ```
 
+## 获取股票代码
+
+```python
+from qdata_quote import get_stock_codes
+
+# 读取缓存（如果没有缓存则自动获取）
+codes = get_stock_codes()
+
+# 强制从网络刷新
+codes = get_stock_codes(refresh=True)
+```
+
+缓存文件位置：`~/.qdata_quote/stock_codes.json`
+
 ## 数据源
 
 支持两个数据源，通过 `source` 参数指定：
@@ -109,18 +124,18 @@ df = service.get_real_sync(["000001"], source="tencent")
 ```python
 # 同步
 with QuoteService() as service:
-    service.set_stock_codes(codes)
+    service.set_stock_codes(get_stock_codes())
     df = service.get_all_sync()
 
 # 异步
 async with QuoteService() as service:
-    service.set_stock_codes(codes)
+    service.set_stock_codes(get_stock_codes())
     df = await service.get_all()
 ```
 
 ## 性能对比
 
-与 easyquotation 对比（5610 只股票）：
+与 easyquotation 对比（5500+ 只股票）：
 
 | 数据源 | easyquotation | qdata_quote sync | qdata_quote async |
 |--------|--------------|-------------------|-------------------|

{qdata_quote-0.2.2 → qdata_quote-0.3.0}/README.md

@@ -11,12 +11,13 @@ pip install qdata-quote
 ## 快速开始
 
 ```python
-from qdata_quote import QuoteService
+from qdata_quote import QuoteService, get_stock_codes
 
-service = QuoteService()
+# 获取全市场股票代码（首次从网络获取约 4 秒，后续读缓存 <1ms）
+codes = get_stock_codes()
 
-# 设置全市场股票代码（必需）
-service.set_stock_codes(["000001", "600000", "600036", "300750"])
+service = QuoteService()
+service.set_stock_codes(codes)
 
 # 同步获取指定股票行情
 df = service.get_real_sync(["000001", "600000"])
@@ -31,6 +32,20 @@ df = asyncio.run(service.get_real(["000001", "600000"]))
 df_all = asyncio.run(service.get_all())
 ```
 
+## 获取股票代码
+
+```python
+from qdata_quote import get_stock_codes
+
+# 读取缓存（如果没有缓存则自动获取）
+codes = get_stock_codes()
+
+# 强制从网络刷新
+codes = get_stock_codes(refresh=True)
+```
+
+缓存文件位置：`~/.qdata_quote/stock_codes.json`
+
 ## 数据源
 
 支持两个数据源，通过 `source` 参数指定：
@@ -95,18 +110,18 @@ df = service.get_real_sync(["000001"], source="tencent")
 ```python
 # 同步
 with QuoteService() as service:
-    service.set_stock_codes(codes)
+    service.set_stock_codes(get_stock_codes())
     df = service.get_all_sync()
 
 # 异步
 async with QuoteService() as service:
-    service.set_stock_codes(codes)
+    service.set_stock_codes(get_stock_codes())
     df = await service.get_all()
 ```
 
 ## 性能对比
 
-与 easyquotation 对比（5610 只股票）：
+与 easyquotation 对比（5500+ 只股票）：
 
 | 数据源 | easyquotation | qdata_quote sync | qdata_quote async |
 |--------|--------------|-------------------|-------------------|

{qdata_quote-0.2.2 → qdata_quote-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "qdata-quote"
-version = "0.2.2"
+version = "0.3.0"
 description = "实时行情采集服务，支持新浪和腾讯数据源"
 readme = "README.md"
 requires-python = ">=3.10"

qdata_quote-0.3.0/src/qdata_quote/__init__.py

@@ -0,0 +1,4 @@
+from .service import QuoteService
+from .stock_codes import get_stock_codes
+
+__all__ = ["QuoteService", "get_stock_codes"]

qdata_quote-0.3.0/src/qdata_quote/stock_codes.py

@@ -0,0 +1,74 @@
+"""全市场股票代码获取与缓存"""
+
+from __future__ import annotations
+
+import json
+import os
+from concurrent.futures import ThreadPoolExecutor
+from pathlib import Path
+
+import requests
+
+_CACHE_PATH = Path.home() / ".qdata_quote" / "stock_codes.json"
+
+
+def get_stock_codes(refresh: bool = False) -> list[str]:
+    """获取全市场 A 股股票代码列表
+
+    首次调用从新浪 API 获取并缓存到本地，后续直接读取缓存。
+
+    Args:
+        refresh: 是否强制从网络刷新，默认 False 读取缓存
+
+    Returns:
+        股票代码列表，如 ["000001", "000002", "600000", ...]
+    """
+    if not refresh and _CACHE_PATH.exists():
+        with open(_CACHE_PATH) as f:
+            return json.load(f)
+
+    codes = _fetch_from_sina()
+    _CACHE_PATH.parent.mkdir(parents=True, exist_ok=True)
+    with open(_CACHE_PATH, "w") as f:
+        json.dump(codes, f)
+    return codes
+
+
+def _fetch_from_sina() -> list[str]:
+    """从新浪 API 并发分页获取全部 A 股代码"""
+    session = requests.Session()
+    session.headers.update(
+        {
+            "Referer": "https://finance.sina.com.cn/",
+            "User-Agent": (
+                "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+                "AppleWebKit/537.36 (KHTML, like Gecko) "
+                "Chrome/120.0.0.0 Safari/537.36"
+            ),
+        }
+    )
+    base_url = (
+        "https://vip.stock.finance.sina.com.cn/quotes_service/"
+        "api/json_v2.php/Market_Center.getHQNodeData?"
+        "page={}&num=80&sort=symbol&asc=1&node=hs_a&symbol="
+    )
+
+    def fetch_page(page):
+        try:
+            r = session.get(base_url.format(page), timeout=10)
+            data = json.loads(r.text)
+            return [s["code"] for s in data] if data else []
+        except Exception:
+            return []
+
+    # 80 页覆盖约 6400 只，超过当前 A 股总数
+    with ThreadPoolExecutor(max_workers=20) as pool:
+        results = pool.map(fetch_page, range(1, 81))
+
+    all_codes = []
+    for codes in results:
+        all_codes.extend(codes)
+
+    session.close()
+    # 去重保序
+    return list(dict.fromkeys(all_codes))

{qdata_quote-0.2.2 → qdata_quote-0.3.0}/uv.lock

@@ -1103,7 +1103,7 @@ wheels = [
 
 [[package]]
 name = "qdata-quote"
-version = "0.2.0"
+version = "0.2.2"
 source = { editable = "." }
 dependencies = [
     { name = "aiohttp" },

qdata_quote-0.2.2/src/qdata_quote/__init__.py

@@ -1,3 +0,0 @@
-from .service import QuoteService
-
-__all__ = ["QuoteService"]