Merge pull request #223 from QuantStrategyLab/codex/sync-qsl-requires-map-20260710

Pigbibi · web-flow · commit 56a03364d517 · 2026-07-11T00:36:11.000+08:00
fix(qpk): sync QSL requires-map pin
diff --git a/scripts/open_downstream_qpk_pin_prs.py b/scripts/open_downstream_qpk_pin_prs.py
@@ -22,6 +22,13 @@
 QSL_QPK_REQUIREMENT_RE = re.compile(
     r"(quant-platform-kit\s*@\s*git\+https://github\.com/QuantStrategyLab/QuantPlatformKit\.git@)[a-f0-9]{40}"
 )
+QSL_REQUIRES_TABLE_RE = re.compile(
+    r"(?ms)^[ \t]*(?P<header>\[qsl\.requires\][^\n]*\n)(?P<body>.*?)(?=^[ \t]*\[|\Z)"
+)
+QSL_QPK_REQUIRES_MAP_RE = re.compile(
+    r"""(^[ \t]*(?:["']quant-platform-kit["']|quant-platform-kit|["']quant_platform_kit["']|quant_platform_kit)\s*=\s*)(['"])[a-f0-9]{40}(\2)""",
+    re.MULTILINE,
+)
 
 
 @dataclass(frozen=True)
@@ -65,7 +72,17 @@ def update_qsl_compat_qpk_pin(repo_dir: Path, qpk_sha: str) -> bool:
         return False
 
     original = qsl_path.read_text(encoding="utf-8")
-    updated, replacements = QSL_QPK_REQUIREMENT_RE.subn(rf"\g<1>{qpk_sha}", original)
+    updated, requirement_replacements = QSL_QPK_REQUIREMENT_RE.subn(rf"\g<1>{qpk_sha}", original)
+    requires_map_replacements = 0
+
+    def update_requires_table(match: re.Match[str]) -> str:
+        nonlocal requires_map_replacements
+        body, replacements = QSL_QPK_REQUIRES_MAP_RE.subn(rf"\g<1>\g<2>{qpk_sha}\g<3>", match.group("body"))
+        requires_map_replacements += replacements
+        return match.group("header") + body
+
+    updated = QSL_REQUIRES_TABLE_RE.sub(update_requires_table, updated)
+    replacements = requirement_replacements + requires_map_replacements
     if replacements == 0 or updated == original:
         return False
 
diff --git a/tests/test_qpk_pin_consistency.py b/tests/test_qpk_pin_consistency.py
@@ -81,6 +81,43 @@ def test_update_qsl_compat_qpk_pin_rewrites_only_qpk_requirement(self) -> None:
         self.assertIn(f"QuantPlatformKit.git@{TARGET}", updated)
         self.assertIn(f"UsEquityStrategies.git@{STALE}", updated)
 
+    def test_update_qsl_compat_qpk_pin_rewrites_qsl_requires_map(self) -> None:
+        with tempfile.TemporaryDirectory() as tmp:
+            qsl_path = Path(tmp) / "qsl.toml"
+            qsl_path.write_text(
+                "[qsl.requires]\n"
+                f'quant_platform_kit = "{STALE}"\n'
+                f'quant-platform-kit = "{STALE}"\n'
+                f'crypto_strategies = "{STALE}"\n'
+                "\n[runtime]\n"
+                f'quant_platform_kit = "{STALE}"\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'quant_platform_kit = "{TARGET}"', updated)
+        self.assertIn(f'quant-platform-kit = "{TARGET}"', updated)
+        self.assertIn(f'crypto_strategies = "{STALE}"', updated)
+        self.assertIn(f'[runtime]\nquant_platform_kit = "{STALE}"', updated)
+
+    def test_update_qsl_compat_qpk_pin_rewrites_single_quoted_qsl_requires_map(self) -> None:
+        with tempfile.TemporaryDirectory() as tmp:
+            qsl_path = Path(tmp) / "qsl.toml"
+            qsl_path.write_text(
+                "[qsl.requires]\n"
+                f"quant_platform_kit = '{STALE}'\n"
+                f"'quant-platform-kit' = '{STALE}'\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"quant_platform_kit = '{TARGET}'", updated)
+        self.assertIn(f"'quant-platform-kit' = '{TARGET}'", updated)
+
 
 if __name__ == "__main__":
     unittest.main()
