qdata-quote 0.3.2__tar.gz → 0.3.4__tar.gz

{qdata_quote-0.3.2 → qdata_quote-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qdata-quote
-Version: 0.3.2
+Version: 0.3.4
 Summary: 实时行情采集服务，支持新浪和腾讯数据源
 License-Expression: MIT
 Requires-Python: >=3.10
@@ -25,13 +25,10 @@ pip install qdata-quote
 ## 快速开始
 
 ```python
-from qdata_quote import QuoteService, get_stock_codes
-
-# 获取全市场股票代码（首次从网络获取约 4 秒，后续读缓存 <1ms）
-codes = get_stock_codes()
+from qdata_quote import QuoteService
 
+# 默认加载全市场股票列表（初始化时自动从网络刷新）
 service = QuoteService()
-service.set_stock_codes(codes)
 
 # 同步获取指定股票行情
 df = service.get_real_sync(["000001", "600000"])
@@ -46,6 +43,21 @@ df = asyncio.run(service.get_real(["000001", "600000"]))
 df_all = asyncio.run(service.get_all())
 ```
 
+## 自定义股票列表
+
+```python
+from qdata_quote import QuoteService
+
+# 传入自定义股票列表（不会自动刷新）
+service = QuoteService(codes=["000001", "600000", "300750"])
+
+# 随时替换股票列表（切换为自定义模式，不再接受 refresh）
+service.set_stock_codes(["000001", "600000"])
+
+# 刷新全市场股票列表（仅在使用默认列表时生效）
+service.refresh_stock_codes()
+```
+
 ## 获取股票代码
 
 ```python
@@ -124,12 +136,10 @@ df = service.get_real_sync(["000001"], source="tencent")
 ```python
 # 同步
 with QuoteService() as service:
-    service.set_stock_codes(get_stock_codes())
     df = service.get_all_sync()
 
 # 异步
 async with QuoteService() as service:
-    service.set_stock_codes(get_stock_codes())
     df = await service.get_all()
 ```
 

{qdata_quote-0.3.2 → qdata_quote-0.3.4}/README.md

@@ -11,13 +11,10 @@ pip install qdata-quote
 ## 快速开始
 
 ```python
-from qdata_quote import QuoteService, get_stock_codes
-
-# 获取全市场股票代码（首次从网络获取约 4 秒，后续读缓存 <1ms）
-codes = get_stock_codes()
+from qdata_quote import QuoteService
 
+# 默认加载全市场股票列表（初始化时自动从网络刷新）
 service = QuoteService()
-service.set_stock_codes(codes)
 
 # 同步获取指定股票行情
 df = service.get_real_sync(["000001", "600000"])
@@ -32,6 +29,21 @@ df = asyncio.run(service.get_real(["000001", "600000"]))
 df_all = asyncio.run(service.get_all())
 ```
 
+## 自定义股票列表
+
+```python
+from qdata_quote import QuoteService
+
+# 传入自定义股票列表（不会自动刷新）
+service = QuoteService(codes=["000001", "600000", "300750"])
+
+# 随时替换股票列表（切换为自定义模式，不再接受 refresh）
+service.set_stock_codes(["000001", "600000"])
+
+# 刷新全市场股票列表（仅在使用默认列表时生效）
+service.refresh_stock_codes()
+```
+
 ## 获取股票代码
 
 ```python
@@ -110,12 +122,10 @@ df = service.get_real_sync(["000001"], source="tencent")
 ```python
 # 同步
 with QuoteService() as service:
-    service.set_stock_codes(get_stock_codes())
     df = service.get_all_sync()
 
 # 异步
 async with QuoteService() as service:
-    service.set_stock_codes(get_stock_codes())
     df = await service.get_all()
 ```
 

{qdata_quote-0.3.2 → qdata_quote-0.3.4}/pyproject.toml

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

{qdata_quote-0.3.2 → qdata_quote-0.3.4}/src/qdata_quote/service.py

@@ -32,7 +32,12 @@ class QuoteService:
     """
 
     def __init__(self, codes: list[str] | None = None):
-        self._stock_codes: list[str] = codes if codes is not None else get_stock_codes()
+        # 标记是否使用自定义列表：自定义列表时 refresh 不覆盖
+        self._use_default_codes = codes is None
+        if self._use_default_codes:
+            self._stock_codes = get_stock_codes(refresh=True)
+        else:
+            self._stock_codes = codes
         self._sources = {
             "sina": SinaSource(),
             "tencent": TencentSource(),
@@ -42,11 +47,18 @@ class QuoteService:
         self._default_source: SourceName = "sina"
 
     def set_stock_codes(self, codes: list[str]) -> None:
-        """替换股票代码列表"""
+        """替换股票代码列表（切换为自定义模式，不再接受 refresh）"""
+        self._use_default_codes = False
         self._stock_codes = codes
 
     def refresh_stock_codes(self) -> None:
-        """从网络刷新全市场股票代码列表并更新"""
+        """从网络刷新全市场股票代码列表
+
+        仅在使用默认列表时生效；通过构造方法或 set_stock_codes
+        指定了自定义列表后，调用此方法不做任何操作。
+        """
+        if not self._use_default_codes:
+            return
         self._stock_codes = get_stock_codes(refresh=True)
 
     # --- 异步接口（aiohttp） ---

{qdata_quote-0.3.2 → qdata_quote-0.3.4}/tests/test_service.py

@@ -30,11 +30,35 @@ class TestQuoteServiceInit:
         assert s._stock_codes == []
 
 
-class TestSetStockCodes:
+class TestStockCodesManagement:
     def test_set_codes(self, service):
         service.set_stock_codes(["000001", "600000"])
         assert service._stock_codes == ["000001", "600000"]
 
+    def test_refresh_with_default_codes(self, service):
+        """默认模式下 refresh 可以刷新"""
+        assert service._use_default_codes is True
+        old = service._stock_codes[:]
+        service.refresh_stock_codes()
+        # 刷新后列表不为空（即使内容相同）
+        assert len(service._stock_codes) > 0
+
+    def test_refresh_with_custom_codes_no_op(self):
+        """自定义模式下 refresh 不覆盖"""
+        s = QuoteService(codes=["000001", "600000"])
+        assert s._use_default_codes is False
+        s.refresh_stock_codes()
+        # 列表保持不变，没有被覆盖为全市场列表
+        assert s._stock_codes == ["000001", "600000"]
+
+    def test_set_codes_disables_refresh(self, service):
+        """set_stock_codes 切换为自定义模式"""
+        assert service._use_default_codes is True
+        service.set_stock_codes(["000001"])
+        assert service._use_default_codes is False
+        service.refresh_stock_codes()
+        assert service._stock_codes == ["000001"]
+
 
 class TestQuoteServiceSync:
     """同步接口集成测试（需要网络）"""

{qdata_quote-0.3.2 → qdata_quote-0.3.4}/uv.lock

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