Merge pull request #221 from QuantStrategyLab/codex/sync-qsl-compat-pin-20260710

fix(qpk): sync QSL compatibility pin metadata

Author: Pigbibi

scripts/open_downstream_qpk_pin_prs.py

@@ -5,6 +5,7 @@
 
 import argparse
 import os
+import re
 import subprocess
 import sys
 import tempfile
@@ -18,6 +19,11 @@
 from check_qpk_pin_consistency import get_qpk_pin_sha
 
 
+QSL_QPK_REQUIREMENT_RE = re.compile(
+    r"(quant-platform-kit\s*@\s*git\+https://github\.com/QuantStrategyLab/QuantPlatformKit\.git@)[a-f0-9]{40}"
+)
+
+
 @dataclass(frozen=True)
 class RepoSpec:
     name: str
@@ -53,13 +59,28 @@ def maybe_run_uv_lock(repo_dir: Path) -> bool:
     return True
 
 
+def update_qsl_compat_qpk_pin(repo_dir: Path, qpk_sha: str) -> bool:
+    qsl_path = repo_dir / "qsl.toml"
+    if not qsl_path.is_file():
+        return False
+
+    original = qsl_path.read_text(encoding="utf-8")
+    updated, replacements = QSL_QPK_REQUIREMENT_RE.subn(rf"\g<1>{qpk_sha}", original)
+    if replacements == 0 or updated == original:
+        return False
+
+    qsl_path.write_text(updated, encoding="utf-8")
+    return True
+
+
 def update_repo(repo_dir: Path, qpk_pin: Path) -> bool:
     script = SCRIPT_ROOT / "check_qpk_pin_consistency.py"
     run(
         ["python3", str(script), "--root", str(repo_dir), "--pin-file", str(qpk_pin), "--fix"],
         cwd=repo_dir,
     )
     maybe_run_uv_lock(repo_dir)
+    update_qsl_compat_qpk_pin(repo_dir, get_qpk_pin_sha(pin_file=qpk_pin))
     return has_changes(repo_dir)
 
 

tests/test_qpk_pin_consistency.py

@@ -9,6 +9,7 @@
     extract_qpk_shas,
     extract_override_qpk_sha,
 )
+from scripts.open_downstream_qpk_pin_prs import update_qsl_compat_qpk_pin
 
 
 TARGET = "8378e939d9324ea63a0f45c9f21ba0e2eeb1cfff"
@@ -61,6 +62,25 @@ def test_override_must_match_pin(self) -> None:
         override = extract_override_qpk_sha(pyproject)
         self.assertEqual("1111111111111111111111111111111111111111", override)
 
+    def test_update_qsl_compat_qpk_pin_rewrites_only_qpk_requirement(self) -> None:
+        with tempfile.TemporaryDirectory() as tmp:
+            qsl_path = Path(tmp) / "qsl.toml"
+            qsl_path.write_text(
+                "[compat]\n"
+                "requires = [\n"
+                f'  "pandas>=2.0",\n'
+                f'  "quant-platform-kit @ git+https://github.com/QuantStrategyLab/QuantPlatformKit.git@{STALE}",\n'
+                f'  "us-equity-strategies @ git+https://github.com/QuantStrategyLab/UsEquityStrategies.git@{STALE}",\n'
+                "]\n",
+                encoding="utf-8",
+            )
+
+            self.assertTrue(update_qsl_compat_qpk_pin(Path(tmp), TARGET))
+            updated = qsl_path.read_text(encoding="utf-8")
+
+        self.assertIn(f"QuantPlatformKit.git@{TARGET}", updated)
+        self.assertIn(f"UsEquityStrategies.git@{STALE}", updated)
+
 
 if __name__ == "__main__":
     unittest.main()