quantcli 0.2.1__tar.gz → 0.2.2__tar.gz

{quantcli-0.2.1/quantcli.egg-info → quantcli-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantcli
-Version: 0.2.1
+Version: 0.2.2
 Summary: 面向AI的多因子量化选股策略挖掘工具，AI Agent 友好 CLI
 Author-email: QuantCLI Team <quantcli@example.com>
 Project-URL: repository, https://github.com/wumu2013/quantcli

{quantcli-0.2.1 → quantcli-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "quantcli"
-version = "0.2.1"
+version = "0.2.2"
 description = "面向AI的多因子量化选股策略挖掘工具，AI Agent 友好 CLI"
 readme = "README.md"
 requires-python = ">=3.9"

{quantcli-0.2.1 → quantcli-0.2.2}/quantcli/cli.py

@@ -111,6 +111,9 @@ def data(ctx):
     help="数据源 (akshare/baostock/mixed/mysql)",
 )
 @click.option("--use-cache/--no-cache", default=True, help="是否使用缓存")
+@click.option(
+    "--cache-raw/--no-cache-raw", default=False, help="是否缓存原始数据到parquet"
+)
 @click.option(
     "--output",
     "-o",
@@ -119,7 +122,7 @@ def data(ctx):
     help="输出文件路径 (如: data.csv)",
 )
 @click.pass_context
-def data_fetch(ctx, symbol, start, end, source, use_cache, output):
+def data_fetch(ctx, symbol, start, end, source, use_cache, cache_raw, output):
     """获取股票日线数据
 
     股票代码:
@@ -131,6 +134,7 @@ def data_fetch(ctx, symbol, start, end, source, use_cache, output):
       quantcli data fetch 000001 --start 2025-01-01 --end 2025-12-31
       quantcli data fetch 600519 --start 2025-01-01 --source mysql
       quantcli data fetch 000001 --start 2025-01-01 -o data.csv
+      quantcli data fetch 000001 --start 2025-01-01 --cache-raw
     """
     if end is None:
         end = today()
@@ -139,7 +143,7 @@ def data_fetch(ctx, symbol, start, end, source, use_cache, output):
         f"Fetching {symbol} data from {format_date(start)} to {format_date(end)}..."
     )
 
-    config = DataConfig(source=source)
+    config = DataConfig(source=source, cache_raw=cache_raw)
     dm = DataManager(config)
 
     try:

{quantcli-0.2.1 → quantcli-0.2.2}/quantcli/core/data.py

@@ -30,6 +30,7 @@ logger = get_logger(__name__)
 class DataConfig:
     """数据管理配置"""
 
+    cache_raw: bool = False
     source: str = "akshare"
     cache_dir: str = "./data"
     parallel: int = 4
@@ -141,8 +142,7 @@ class DataManager:
             logger.warning(f"No data for {symbol}")
             return df
 
-        # 缓存数据
-        if use_cache:
+        if use_cache and self.config.cache_raw:
             self._save_to_cache(cache_file, df, f"stock_daily_{symbol}")
 
         return df
@@ -257,7 +257,7 @@ class DataManager:
 
         df = self.datasource.get_index_daily(symbol, start_date, end_date)
 
-        if use_cache and not df.empty:
+        if use_cache and self.config.cache_raw and not df.empty:
             self._save_to_cache(cache_file, df, f"index_daily_{symbol}")
 
         return df

{quantcli-0.2.1 → quantcli-0.2.2/quantcli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantcli
-Version: 0.2.1
+Version: 0.2.2
 Summary: 面向AI的多因子量化选股策略挖掘工具，AI Agent 友好 CLI
 Author-email: QuantCLI Team <quantcli@example.com>
 Project-URL: repository, https://github.com/wumu2013/quantcli

{quantcli-0.2.1 → quantcli-0.2.2}/tests/test_data_manager.py

@@ -52,7 +52,7 @@ class TestGetDailyCacheLogic:
         self, temp_cache_dir, mock_datasource
     ):
         """缓存存在且完整 → 直接返回缓存"""
-        config = DataConfig(cache_dir=temp_cache_dir, source="mock")
+        config = DataConfig(cache_dir=temp_cache_dir, source="mock", cache_raw=True)
         dm = DataManager(config)
         dm._datasource = mock_datasource
 
@@ -95,7 +95,7 @@ class TestGetDailyCacheLogic:
 
     def test_cache_incomplete_fetches_full_data(self, temp_cache_dir, mock_datasource):
         """缓存不完整 → 全量拉取"""
-        config = DataConfig(cache_dir=temp_cache_dir, source="mock")
+        config = DataConfig(cache_dir=temp_cache_dir, source="mock", cache_raw=True)
         dm = DataManager(config)
         dm._datasource = mock_datasource
 
@@ -138,7 +138,7 @@ class TestGetDailyCacheLogic:
 
     def test_no_cache_fetches_full_data(self, temp_cache_dir, mock_datasource):
         """缓存不存在 → 全量拉取"""
-        config = DataConfig(cache_dir=temp_cache_dir, source="mock")
+        config = DataConfig(cache_dir=temp_cache_dir, source="mock", cache_raw=True)
         dm = DataManager(config)
         dm._datasource = mock_datasource
 
@@ -165,7 +165,7 @@ class TestGetDailyCacheLogic:
 
     def test_no_calendar_fetches_full_data(self, temp_cache_dir, mock_datasource):
         """交易日历不可用 → 直接拉取，不使用缓存"""
-        config = DataConfig(cache_dir=temp_cache_dir, source="mock")
+        config = DataConfig(cache_dir=temp_cache_dir, source="mock", cache_raw=True)
         dm = DataManager(config)
         dm._datasource = mock_datasource
 
@@ -200,7 +200,7 @@ class TestGetDailyCacheLogic:
         self, temp_cache_dir, mock_datasource
     ):
         """交易日历不可用 + 无缓存 → 全量拉取"""
-        config = DataConfig(cache_dir=temp_cache_dir, source="mock")
+        config = DataConfig(cache_dir=temp_cache_dir, source="mock", cache_raw=True)
         dm = DataManager(config)
         dm._datasource = mock_datasource
 
@@ -222,7 +222,9 @@ class TestGetDailyBatch:
 
     def test_batch_uses_parallel_when_enabled(self, temp_cache_dir, mock_datasource):
         """批量获取启用并行"""
-        config = DataConfig(cache_dir=temp_cache_dir, source="mock", parallel=2)
+        config = DataConfig(
+            cache_dir=temp_cache_dir, source="mock", parallel=2, cache_raw=True
+        )
         dm = DataManager(config)
         dm._datasource = mock_datasource
 
@@ -240,7 +242,7 @@ class TestCacheCleanup:
 
     def test_clear_cache_removes_files(self, temp_cache_dir, mock_datasource):
         """清理缓存删除文件"""
-        config = DataConfig(cache_dir=temp_cache_dir, source="mock")
+        config = DataConfig(cache_dir=temp_cache_dir, source="mock", cache_raw=True)
         dm = DataManager(config)
         dm._datasource = mock_datasource