deeptrade-quant 0.4.0__tar.gz → 0.4.1__tar.gz

{deeptrade_quant-0.4.0 → deeptrade_quant-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: deeptrade-quant
-Version: 0.4.0
+Version: 0.4.1
 Summary: LLM-driven A-share (Shanghai/Shenzhen main board) stock screening CLI
 Project-URL: Homepage, https://github.com/ty19880929/deeptrade
 Project-URL: Repository, https://github.com/ty19880929/deeptrade

{deeptrade_quant-0.4.0 → deeptrade_quant-0.4.1}/deeptrade/__init__.py

@@ -2,5 +2,5 @@
 
 from __future__ import annotations
 
-__version__ = "0.4.0"
+__version__ = "0.4.1"
 __all__ = ["__version__"]

deeptrade_quant-0.4.1/deeptrade/core/migrations/core/20260512_001_drop_legacy_tushare_cache.sql

@@ -0,0 +1,10 @@
+-- v0.4.1 — drop legacy tushare_cache_blob entries.
+--
+-- Pre-v0.4.1 payloads were stored as a bare JSON array, then read back via
+-- pd.read_json which triggered a pandas FutureWarning on string columns whose
+-- names matched its date heuristic (trade_date, cal_date, ...). The cache
+-- read/write path now wraps payloads as {version:1, schema:{...}, data:[...]}
+-- and restores dtypes explicitly. Old rows are incompatible with that reader,
+-- so wipe the table; TushareClient._ensure_cache_table re-creates it lazily on
+-- the next write.
+DROP TABLE IF EXISTS tushare_cache_blob;

{deeptrade_quant-0.4.0 → deeptrade_quant-0.4.1}/deeptrade/core/tushare_client.py

@@ -19,7 +19,6 @@ Cache class buckets:
 from __future__ import annotations
 
 import hashlib
-import io
 import json
 import logging
 import threading
@@ -896,6 +895,15 @@ class TushareClient:
             ")"
         )
 
+    # Payload wrapper format (v0.4.1+):
+    #   {"version": 1, "schema": {col: dtype_str}, "data": [...records...]}
+    # Bypasses pd.read_json's date-column heuristic (which warns on string
+    # columns named like dates — trade_date / cal_date / ann_date) and
+    # restores dtypes explicitly from the recorded schema. Pre-v0.4.1 rows
+    # were a bare records array; those are wiped by core migration
+    # 20260512_001_drop_legacy_tushare_cache.sql, so no legacy branch here.
+    _CACHE_PAYLOAD_VERSION = 1
+
     def _write_cached(
         self,
         api_name: str,
@@ -906,7 +914,14 @@ class TushareClient:
         self._ensure_cache_table()
         body = json.dumps(params, sort_keys=True, default=str)
         h = hashlib.sha256(body.encode("utf-8")).hexdigest()
-        payload = df.to_json(orient="records", date_format="iso")
+        records_json = df.to_json(orient="records", date_format="iso")
+        payload = json.dumps(
+            {
+                "version": self._CACHE_PAYLOAD_VERSION,
+                "schema": {col: str(dt) for col, dt in df.dtypes.items()},
+                "data": json.loads(records_json) if records_json else [],
+            }
+        )
         with self._db.transaction():
             self._db.execute(
                 "DELETE FROM tushare_cache_blob "
@@ -937,12 +952,34 @@ class TushareClient:
         )
         if row is None:
             return pd.DataFrame()
-        df = pd.read_json(io.StringIO(row[0]), orient="records")
+        wrapper = json.loads(row[0])
+        df = self._restore_cached_frame(wrapper)
         if fields:
             cols = [c.strip() for c in fields.split(",") if c.strip() in df.columns]
             df = df[cols]
         return df
 
+    @classmethod
+    def _restore_cached_frame(cls, wrapper: dict[str, Any]) -> pd.DataFrame:
+        schema: dict[str, str] = wrapper["schema"]
+        data: list[dict[str, Any]] = wrapper["data"]
+        df = pd.DataFrame.from_records(data, columns=list(schema.keys()))
+        for col, dtype_str in schema.items():
+            if col not in df.columns:
+                continue
+            if dtype_str.startswith("datetime"):
+                df[col] = pd.to_datetime(df[col], errors="coerce")
+            elif dtype_str == "object":
+                continue
+            else:
+                try:
+                    df[col] = df[col].astype(dtype_str)
+                except (TypeError, ValueError):
+                    # Best-effort: if a numeric/bool column can't be coerced
+                    # back (e.g. all-null), leave the inferred dtype.
+                    pass
+        return df
+
 
 # ---------------------------------------------------------------------------
 # Fallback predicate (DESIGN §13.2 + S4)

{deeptrade_quant-0.4.0 → deeptrade_quant-0.4.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "deeptrade-quant"
-version = "0.4.0"
+version = "0.4.1"
 description = "LLM-driven A-share (Shanghai/Shenzhen main board) stock screening CLI"
 readme = "README.md"
 requires-python = ">=3.11"

{deeptrade_quant-0.4.0 → deeptrade_quant-0.4.1}/tests/core/test_db.py

@@ -50,9 +50,9 @@ def test_init_is_idempotent(monkeypatch: pytest.MonkeyPatch, tmp_path: Path) ->
 def test_apply_core_migrations_records_version(fresh_db: Database) -> None:
     """Framework owns no business tables."""
     applied = apply_core_migrations(fresh_db)
-    assert applied == ["20260509_001"]
+    assert applied == ["20260509_001", "20260512_001"]
     rows = fresh_db.fetchall("SELECT version FROM schema_migrations ORDER BY version")
-    assert tuple(rows) == (("20260509_001",),)
+    assert tuple(rows) == (("20260509_001",), ("20260512_001",))
 
 
 def test_apply_core_migrations_skips_applied_versions(fresh_db: Database) -> None:

{deeptrade_quant-0.4.0 → deeptrade_quant-0.4.1}/tests/core/test_tushare_client.py

@@ -5,6 +5,7 @@
 
 from __future__ import annotations
 
+import warnings
 from datetime import datetime, timedelta
 from pathlib import Path
 
@@ -452,3 +453,36 @@ def test_cache_hit_requires_payload_present(
     client.call("limit_list_d", trade_date="20260427")
     # Should hit transport (because no payload despite state=ok)
     assert len(transport.calls) == 1
+
+
+# ---------------------------------------------------------------------------
+# Cache payload round-trip — dtypes preserved, no pandas FutureWarning
+# ---------------------------------------------------------------------------
+
+
+def test_cache_payload_round_trip_preserves_dtypes(client: TushareClient) -> None:
+    """v0.4.1 wrapper format restores dtypes explicitly so plugins don't see
+    pandas' read_json date-heuristic warnings on tushare-style date columns."""
+    df = pd.DataFrame(
+        {
+            "trade_date": ["20260427", "20260428"],  # YYYYMMDD strings — triggers heuristic
+            "ts_code": ["000001.SZ", "000002.SZ"],
+            "close": [12.5, 14.7],
+            "vol": [1000, 2000],
+            "is_st": [True, False],
+            "ann_dt": pd.to_datetime(["2026-04-27", "2026-04-28"]),  # real datetime64[ns]
+        }
+    )
+    api, td, params = "limit_list_d", "20260427", {"trade_date": "20260427"}
+
+    client._write_cached(api, td, params, df)  # noqa: SLF001 — exercising internal contract
+
+    with warnings.catch_warnings():
+        warnings.simplefilter("error", FutureWarning)
+        out = client._read_cached(api, td, params, fields=None)  # noqa: SLF001
+
+    assert list(out.columns) == list(df.columns)
+    for col in df.columns:
+        assert out[col].dtype == df[col].dtype, f"dtype drift on {col}: {out[col].dtype}"
+    assert out["trade_date"].tolist() == ["20260427", "20260428"]
+    assert (out["ann_dt"] == df["ann_dt"]).all()