wudixzy stack docker build test: 2615-line qwen3_5.py + 12 prebuilt .so + 251-line patch_ops.sh

qwen3_6_scripts/patch_utils.py

@@ -0,0 +1,81 @@
+from __future__ import annotations
+
+import importlib.util
+import pathlib
+import shlex
+from typing import Iterable, Optional, Sequence, Tuple
+
+
+def package_root(pkg: str) -> pathlib.Path:
+    spec = importlib.util.find_spec(pkg)
+    if spec is None:
+        raise RuntimeError(f"package not found: {pkg}")
+    if not spec.submodule_search_locations:
+        raise RuntimeError(f"package has no package root: {pkg}")
+    return pathlib.Path(next(iter(spec.submodule_search_locations))).resolve()
+
+
+def ensure_file(path: pathlib.Path) -> pathlib.Path:
+    if not path.is_file():
+        raise FileNotFoundError(str(path))
+    return path
+
+
+def ensure_dir(path: pathlib.Path) -> pathlib.Path:
+    if not path.is_dir():
+        raise FileNotFoundError(str(path))
+    return path
+
+
+def replace_once(path: pathlib.Path,
+                 old: str,
+                 new: str,
+                 *,
+                 required: bool = True,
+                 already_contains: Optional[str] = None) -> bool:
+    path = ensure_file(path)
+    text = path.read_text()
+    marker = already_contains if already_contains is not None else new
+    if marker in text:
+        print(f"[skip] already patched: {path}")
+        return False
+    if old not in text:
+        msg = f"anchor not found in {path}: {old[:120]!r}"
+        if required:
+            raise RuntimeError(msg)
+        print(f"[warn] {msg}")
+        return False
+    path.write_text(text.replace(old, new, 1))
+    print(f"[ok] patched: {path}")
+    return True
+
+
+def replace_one_of(path: pathlib.Path,
+                   replacements: Sequence[Tuple[str, str]],
+                   *,
+                   required: bool = True,
+                   already_contains: Optional[str] = None) -> bool:
+    path = ensure_file(path)
+    text = path.read_text()
+    if already_contains is not None and already_contains in text:
+        print(f"[skip] already patched: {path}")
+        return False
+    for _, new in replacements:
+        if new in text:
+            print(f"[skip] already patched: {path}")
+            return False
+    for old, new in replacements:
+        if old in text:
+            path.write_text(text.replace(old, new, 1))
+            print(f"[ok] patched: {path}")
+            return True
+    anchors = ", ".join(repr(old[:80]) for old, _ in replacements)
+    msg = f"anchor not found in {path}; tried: {anchors}"
+    if required:
+        raise RuntimeError(msg)
+    print(f"[warn] {msg}")
+    return False
+
+
+def shell_env_line(name: str, value: pathlib.Path) -> str:
+    return f"{name}={shlex.quote(str(value))}"