ai 开始测试

Utils/DevDiskImageDeployer.py

@@ -0,0 +1,123 @@
+# support_deployer.py
+import os
+import re
+import shutil
+from pathlib import Path
+from dataclasses import dataclass, field
+from typing import Iterable, Optional
+
+VERSION_RE = re.compile(r"^\d+(?:\.\d+)*$")  # 15 / 15.6 / 16.7 / 16.7.1
+
+def _find_support_root(hint: Optional[Path]) -> Optional[Path]:
+    """
+    1) 优先：显式传入的 hint
+    2) 其次：环境变量 SUPPORT_DDI_DIR
+    3) 再次：从 __file__ 所在目录向上搜索 3 层，找名为 'SupportFiles' 的目录
+    """
+    # 1) 显式参数
+    if hint and hint.exists():
+        return hint.resolve()
+
+    # 2) 环境变量
+    env = os.environ.get("SUPPORT_DDI_DIR")
+    if env:
+        p = Path(env).expanduser()
+        if p.exists():
+            return p.resolve()
+
+    # 3) 向上搜索
+    here = Path(__file__).resolve().parent
+    for i in range(4):  # 当前目录 + 向上 3 层
+        cand = here / "SupportFiles"
+        if cand.exists():
+            return cand.resolve()
+        here = here.parent
+
+    return None
+
+
+@dataclass
+class DevDiskImageDeployer:
+    """
+    同步 SupportFiles/<version>/... 到 ~/.tidevice/device-support/<version>/...
+    - 自动定位 SupportFiles（显式参数/环境变量/向上搜索）
+    - 已存在则跳过，不存在才复制
+    - 详细日志
+    """
+    project_support_root: Optional[Path] = None  # 可不传，自动发现
+    cache_root: Optional[Path] = None           # 默认 ~/.tidevice/device-support
+    verbose: bool = True
+    dry_run: bool = False
+    overwrite: bool = False
+
+    _src_dir: Path = field(init=False, repr=False)
+    _cache_dir: Path = field(init=False, repr=False)
+
+    def __post_init__(self):
+        src = _find_support_root(self.project_support_root)
+        if src is None:
+            raise FileNotFoundError(
+                "未找到 SupportFiles 目录。"
+                "可传入 project_support_root，或设置环境变量 SUPPORT_DDI_DIR，"
+                "或确保在当前文件上层 3 级目录内存在名为 'SupportFiles' 的目录。"
+            )
+        self._src_dir = src
+
+        if self.cache_root is None:
+            self._cache_dir = Path.home() / ".tidevice" / "device-support"
+        else:
+            self._cache_dir = Path(self.cache_root).expanduser().resolve()
+        self._cache_dir.mkdir(parents=True, exist_ok=True)
+
+        if self.verbose:
+            print(f"[INFO] resolved SupportFiles = {self._src_dir}")
+            print(f"[INFO] cache_dir            = {self._cache_dir}")
+
+            # 打印一眼能看见的兄弟目录，防路径误判
+            parent = self._src_dir.parent
+            try:
+                siblings = ", ".join(sorted(p.name for p in parent.iterdir() if p.is_dir()))
+                print(f"[INFO] SupportFiles parent = {parent}")
+                print(f"[INFO] siblings            = {siblings}")
+            except Exception:
+                pass
+
+    def deploy_all(self):
+        candidates = list(self._iter_version_dirs(self._src_dir))
+        if self.verbose:
+            if not candidates:
+                print("[WARN] 未发现任何版本目录（仅匹配 15 / 15.6 / 16.7 / 16.7.1 这种名字）")
+            else:
+                print("[INFO] 发现版本目录：")
+                for c in candidates:
+                    print(f"       - {c.name}")
+
+        copied = skipped = 0
+        for ver_dir in candidates:
+            dst = self._cache_dir / ver_dir.name
+            if dst.exists() and not self.overwrite:
+                if self.verbose:
+                    print(f"[SKIP] 已存在：{dst}")
+                skipped += 1
+                continue
+
+            if dst.exists() and self.overwrite:
+                if self.verbose:
+                    print(f"[INFO] 覆盖模式，删除：{dst}")
+                if not self.dry_run:
+                    shutil.rmtree(dst)
+
+            if self.verbose:
+                print(f"[COPY] {ver_dir}  ->  {dst}")
+            if not self.dry_run:
+                shutil.copytree(ver_dir, dst)
+            copied += 1
+
+        if self.verbose:
+            print(f"[SUMMARY] copied={copied}, skipped={skipped}, total={copied+skipped}")
+
+    # -------- helpers --------
+    def _iter_version_dirs(self, root: Path) -> Iterable[Path]:
+        for p in sorted(root.iterdir()):
+            if p.is_dir() and VERSION_RE.match(p.name):
+                yield p