quantdb-sdk 0.2.2__tar.gz → 0.2.4__tar.gz

{quantdb_sdk-0.2.2 → quantdb_sdk-0.2.4}/CHANGELOG.md

@@ -3,6 +3,16 @@
 所有 notable 变更都会记录在此文件。格式基于 [Keep a Changelog](https://keepachangelog.com/zh-CN/1.1.0/)，
 版本号遵循 [Semantic Versioning](https://semver.org/lang/zh-CN/)。
 
+## [0.2.4] - 2026-07-26
+
+### Fixed
+- **Windows CLI 编码**：移除 `✓/✗` 字符，GBK 控制台可正常执行 `quantdb check`。
+
+## [0.2.3] - 2026-07-26
+
+### Added
+- **财务本地仓库**：`mount_local_dataset()` 支持财务 V1、ETF/可转债与更多日频子类；财务数据以 `Symbol + m_timetag/endDate/time` 业务主键去重。
+
 ## [0.2.2] - 2026-07-26
 
 ### Added

{quantdb_sdk-0.2.2/quantdb_sdk.egg-info → quantdb_sdk-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantdb-sdk
-Version: 0.2.2
+Version: 0.2.4
 Summary: QuantDB 量化数据平台官方 Python SDK
 Author: QuantDB Team
 License: MIT

{quantdb_sdk-0.2.2 → quantdb_sdk-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "quantdb-sdk"
-version = "0.2.2"
+version = "0.2.4"
 description = "QuantDB 量化数据平台官方 Python SDK"
 readme = "README.md"
 license = {text = "MIT"}

{quantdb_sdk-0.2.2 → quantdb_sdk-0.2.4}/quantdb_sdk/__init__.py

@@ -5,7 +5,7 @@ try:
     # 注意：此处用 PyPI 包名（连字符）查找版本号，不是模块名（下划线）
     __version__ = version("quantdb-sdk")
 except ImportError:
-    __version__ = "0.2.2"
+    __version__ = "0.2.4"
 
 from .async_client import AsyncQuantDBClient
 from .client import DuckDBWarehouse, QuantDBClient

{quantdb_sdk-0.2.2 → quantdb_sdk-0.2.4}/quantdb_sdk/__main__.py

@@ -56,7 +56,7 @@ def main() -> int:
 
         if args.command == "check":
             me = client.get_me()
-            print(f"✓ API Key 有效")
+            print("[OK] API Key 有效")
             print(f"  用户: {me.get('username', 'N/A')}")
             print(f"  ID: {me.get('id', 'N/A')}")
             return 0
@@ -81,7 +81,7 @@ def main() -> int:
             return 0
 
     except Exception as e:
-        print(f"✗ 错误: {e}", file=sys.stderr)
+        print(f"[ERROR] {e}", file=sys.stderr)
         return 1
 
     return 0

{quantdb_sdk-0.2.2 → quantdb_sdk-0.2.4}/quantdb_sdk/async_client.py

@@ -37,7 +37,7 @@ class AsyncQuantDBClient:
     ):
         self.api_host = api_host.rstrip("/")
         self.timeout = timeout
-        headers = {"User-Agent": "QuantDB-Python-SDK/0.2.2"}
+        headers = {"User-Agent": "QuantDB-Python-SDK/0.2.4"}
         if api_key:
             headers["X-API-Key"] = api_key
         elif token:

{quantdb_sdk-0.2.2 → quantdb_sdk-0.2.4}/quantdb_sdk/client.py

@@ -48,7 +48,7 @@ class QuantDBClient:
         self.session = requests.Session()
         # User-Agent 中的版本与 pyproject.toml 同步，用于服务端日志归因
         # 维护提示：每次版本号变化必须同步改这里（init 里的 __version__ 走 metadata 自动同步）
-        self.session.headers.update({"User-Agent": "QuantDB-Python-SDK/0.2.2"})
+        self.session.headers.update({"User-Agent": "QuantDB-Python-SDK/0.2.4"})
 
         if api_key:
             self.headers = {"X-API-Key": api_key}
@@ -713,7 +713,17 @@ class QuantDBClient:
         """挂载本地 V1/V2 同构数据集；V2 patch 与 daily 文件均参与去重。"""
         warehouse = self.get_local_warehouse(save_dir)
         root = os.path.abspath(save_dir or default_download_dir()).replace("\\", "/")
-        views = {"daily_forward":"1_kline_data/daily_forward", "valuation":"5_technical_derived/valuation", "technical_indicators":"5_technical_derived/technical_indicators", "market_sentiment":"5_technical_derived/market_sentiment", "features_daily":"6_ml_datasets/features_daily"}
+        views = {
+            "daily_unadjusted": "1_kline_data/daily_unadjusted", "daily_forward": "1_kline_data/daily_forward",
+            "daily_backward": "1_kline_data/daily_backward", "index_daily": "1_kline_data/index_daily",
+            "margin_trading": "2_base_sector/margin_trading", "valuation": "5_technical_derived/valuation",
+            "technical_indicators": "5_technical_derived/technical_indicators", "market_sentiment": "5_technical_derived/market_sentiment",
+            "features_daily": "6_ml_datasets/features_daily", "balance": "3_financial_data/balance",
+            "income": "3_financial_data/income", "cashflow": "3_financial_data/cashflow",
+            "capital": "3_financial_data/capital", "holder_num": "3_financial_data/holder_num",
+            "pershare_index": "3_financial_data/pershare_index", "dividend_factors": "3_financial_data/dividend_factors",
+            "etf_pcf": "4_bond_etf/etf_pcf", "convertible_bond": "4_bond_etf/convertible_bond",
+        }
         if dataset not in views: raise ValidationError(f"不支持的 V2 数据集: {dataset}")
         name = view_name or re.sub(r"[^a-zA-Z0-9_]", "_", dataset)
         state_path = os.path.join(root.replace("/", os.sep), "quantdb_sync.sqlite")
@@ -732,9 +742,13 @@ class QuantDBClient:
         quoted = ", ".join("'" + p.replace("'", "''") + "'" for p in files)
         relation = f"read_parquet([{quoted}], union_by_name=true)"
         columns = {row[0] for row in warehouse.conn.execute(f"DESCRIBE SELECT * FROM {relation}").fetchall()}
-        if {"symbol", "time"}.issubset(columns):
+        symbol_column = "symbol" if "symbol" in columns else ("Symbol" if "Symbol" in columns else None)
+        date_column = next((column for column in ("time", "m_timetag", "endDate", "date") if column in columns), None)
+        if symbol_column and date_column:
             order = "coalesce(release_id, '') DESC" if "release_id" in columns else "0"
-            warehouse.conn.execute(f"CREATE OR REPLACE VIEW {name} AS SELECT * EXCLUDE (rn) FROM (SELECT *, row_number() OVER (PARTITION BY symbol, time ORDER BY {order}) rn FROM {relation}) WHERE rn=1")
+            symbol_sql = '"' + symbol_column.replace('"', '""') + '"'
+            date_sql = '"' + date_column.replace('"', '""') + '"'
+            warehouse.conn.execute(f"CREATE OR REPLACE VIEW {name} AS SELECT * EXCLUDE (rn) FROM (SELECT *, row_number() OVER (PARTITION BY {symbol_sql}, {date_sql} ORDER BY {order}) rn FROM {relation}) WHERE rn=1")
         else:
             warehouse.conn.execute(f"CREATE OR REPLACE VIEW {name} AS SELECT * FROM {relation}")
         warehouse._views[name] = f"{dataset}"

{quantdb_sdk-0.2.2 → quantdb_sdk-0.2.4/quantdb_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantdb-sdk
-Version: 0.2.2
+Version: 0.2.4
 Summary: QuantDB 量化数据平台官方 Python SDK
 Author: QuantDB Team
 License: MIT

{quantdb_sdk-0.2.2 → quantdb_sdk-0.2.4}/tests/test_client.py

@@ -230,3 +230,14 @@ def test_duckdb_warehouse(tmp_path):
     df = wh.query("SELECT * FROM stock_val WHERE val > 15")
     assert len(df) == 1
     assert df.iloc[0]["code"] == "600519.SH"
+
+
+def test_mount_local_financial_dataset_deduplicates_business_key(tmp_path):
+    pytest.importorskip("duckdb")
+    directory = tmp_path / "3_financial_data" / "balance"
+    directory.mkdir(parents=True)
+    pd.DataFrame({"Symbol": ["600000.SH"], "m_timetag": [20260331], "value": [1]}).to_parquet(directory / "a.parquet")
+    pd.DataFrame({"Symbol": ["600000.SH"], "m_timetag": [20260331], "value": [2]}).to_parquet(directory / "b.parquet")
+    client = QuantDBClient(api_host=API_HOST, api_key="test-key")
+    warehouse = client.mount_local_dataset("balance", str(tmp_path))
+    assert len(warehouse.query("SELECT * FROM balance")) == 1