sino-bc-cli 1.0.0-beta.0

package/dist/skills/biz-line-strategy-generate/scripts/cleanup.py

@@ -0,0 +1,87 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""仅清理本次进程创建并登记的目录，所有类型临时文件统一处理。"""
+from __future__ import annotations
+
+import json
+import os
+import stat
+import tempfile
+import uuid
+from dataclasses import dataclass
+from pathlib import Path
+
+MARKER = '.strategy-owner.json'
+
+
+def is_link(path: Path) -> bool:
+    """Windows junction 等重解析点与符号链接都不允许递归跟随。"""
+    info = path.lstat()
+    return stat.S_ISLNK(info.st_mode) or bool(getattr(info, 'st_file_attributes', 0) & 0x400)
+
+
+@dataclass(frozen=True)
+class OwnedDirectory:
+    """创建时保存真实父路径与随机令牌，清理时不能只相信目录名。"""
+    path: Path
+    parent: Path
+    token: str
+
+    @classmethod
+    def create(cls, parent: Path, prefix: str):
+        parent = parent.resolve()
+        parent.mkdir(parents=True, exist_ok=True)
+        path = Path(tempfile.mkdtemp(prefix=prefix, dir=parent))
+        token = uuid.uuid4().hex
+        (path / MARKER).write_text(json.dumps({'token': token, 'path': str(path)}), encoding='utf-8')
+        return cls(path, parent, token)
+
+    def validate(self):
+        """根目录或父路径被替换后拒绝删除，防止越界。"""
+        if is_link(self.path) or self.path.resolve() != self.path or self.path.parent.resolve() != self.parent:
+            raise ValueError(f'清理范围已变化：{self.path}')
+        marker = self.path / MARKER
+        if is_link(marker):
+            raise ValueError('所有权文件不能是链接')
+        data = json.loads(marker.read_text(encoding='utf-8'))
+        if data != {'token': self.token, 'path': str(self.path)}:
+            raise ValueError(f'所有权不匹配：{self.path}')
+
+    def cleanup(self, final: Path | None = None) -> dict:
+        """不按后缀筛选；保留可返回的失败清单，绝不静默忽略残留。"""
+        errors = []
+        try:
+            self.validate()
+            if final is not None and final.resolve().is_relative_to(self.path):
+                raise ValueError('最终 Excel 位于临时目录内，拒绝清理')
+        except Exception as exc:
+            return {'ok': False, 'residual': [{'path': str(self.path), 'error': str(exc)}]}
+
+        def remove(path):
+            try:
+                # 先 lstat 再决定删除方式；对 junction 仅解除链接自身。
+                if is_link(path):
+                    if path.is_symlink():
+                        path.unlink()
+                    else:
+                        os.rmdir(path)
+                elif path.is_dir():
+                    for child in path.iterdir():
+                        remove(child)
+                    path.rmdir()
+                else:
+                    path.unlink()
+            except Exception as exc:
+                errors.append({'path': str(path), 'error': str(exc)})
+
+        # 若某子项未能删除，留下所有权标记供诊断，不伪报清理成功。
+        for child in self.path.iterdir():
+            if child.name != MARKER:
+                remove(child)
+        if not errors:
+            remove(self.path / MARKER)
+            try:
+                self.path.rmdir()
+            except Exception as exc:
+                errors.append({'path': str(self.path), 'error': str(exc)})
+        return {'ok': not errors, 'residual': errors}