devmate 1.0.0__py3-none-any.whl

devmate_toolkit/__init__.py

@@ -0,0 +1,223 @@
+"""
+toolkit — 文件瑞士军刀
+
+包含文件整理(organize)、目录大小排行(size)、缓存清理(cleanup)等实用功能。
+"""
+
+from pathlib import Path
+from typing import Any
+
+from devmate.bus.command import CommandResult
+
+# ── 文件类型分类规则 ────────────────────────────────
+
+CATEGORY_RULES: dict[str, list[str]] = {
+    "图片": [".jpg", ".jpeg", ".png", ".gif", ".bmp", ".svg", ".webp", ".ico"],
+    "文档": [".pdf", ".doc", ".docx", ".xls", ".xlsx", ".ppt", ".pptx", ".txt", ".md", ".csv"],
+    "代码": [".py", ".js", ".ts", ".java", ".go", ".rs", ".c", ".cpp", ".h", ".hpp",
+             ".css", ".scss", ".html", ".json", ".xml", ".yaml", ".yml", ".toml"],
+    "压缩包": [".zip", ".tar", ".gz", ".bz2", ".7z", ".rar"],
+    "视频": [".mp4", ".avi", ".mkv", ".mov", ".flv", ".wmv"],
+    "音频": [".mp3", ".wav", ".flac", ".aac", ".ogg"],
+    "安装包": [".dmg", ".pkg", ".exe", ".msi", ".deb", ".rpm", ".AppImage"],
+}
+
+
+def _get_category(suffix: str) -> str:
+    """根据文件后缀返回分类名"""
+    for category, extensions in CATEGORY_RULES.items():
+        if suffix.lower() in extensions:
+            return category
+    return "其他"
+
+
+def _size_str(bytes_val: int) -> str:
+    """字节数转人类可读大小"""
+    for unit in ("B", "KB", "MB", "GB", "TB"):
+        if bytes_val < 1024:
+            return f"{bytes_val:.1f}{unit}"
+        bytes_val /= 1024
+    return f"{bytes_val:.1f}PB"
+
+
+# ── organize：文件整理 ──────────────────────────────
+
+def organize(source: str, dry_run: bool = False, by: str = "type") -> CommandResult:
+    """按类别整理文件
+
+    按文件类型(by=type)或按修改日期(by=date)将文件分类到子目录。
+    """
+    src = Path(source).expanduser().resolve()
+    if not src.is_dir():
+        return CommandResult(success=False, error=f"目录不存在: {source}")
+
+    stats = {"已整理": 0, "跳过": 0, "错误": 0}
+    details: list[dict[str, Any]] = []
+
+    for item in src.iterdir():
+        if not item.is_file():
+            stats["跳过"] += 1
+            continue
+
+        if by == "type":
+            category = _get_category(item.suffix)
+            target_dir = src / category
+            action_label = category
+        elif by == "date":
+            mtime = item.stat().st_mtime
+            from datetime import datetime
+            date_str = datetime.fromtimestamp(mtime).strftime("%Y-%m")
+            target_dir = src / date_str
+            action_label = date_str
+        else:
+            return CommandResult(success=False, error=f"不支持的整理方式: {by}")
+
+        target = target_dir / item.name
+
+        if target.exists():
+            stats["跳过"] += 1
+            details.append({"file": item.name, "action": "跳过（已存在）"})
+            continue
+
+        if not dry_run:
+            target_dir.mkdir(parents=True, exist_ok=True)
+            try:
+                item.rename(target)
+            except OSError:
+                stats["错误"] += 1
+                continue
+
+        stats["已整理"] += 1
+        details.append({"file": item.name, "action": f"→ {action_label}"})
+
+    msg = f"整理完成: {stats['已整理']} 个文件已整理"
+    if stats["跳过"]:
+        msg += f", {stats['跳过']} 个跳过"
+    if stats["错误"]:
+        msg += f", {stats['错误']} 个错误"
+    if dry_run:
+        msg = f"[预览] 将整理 {stats['已整理']} 个文件" + (
+            f"，{stats['跳过']} 个跳过" if stats["跳过"] else ""
+        )
+
+    return CommandResult(success=True, data={
+        "stats": stats,
+        "details": details,
+        "message": msg,
+        "source": str(src),
+    })
+
+
+# ── size：目录大小排行 ─────────────────────────────
+
+def size(path: str, top: int = 10) -> CommandResult:
+    """统计目录大小排行"""
+    target = Path(path).expanduser().resolve()
+    if not target.is_dir():
+        return CommandResult(success=False, error=f"目录不存在: {path}")
+
+    entries: list[dict[str, Any]] = []
+    total_size = 0
+    file_count = 0
+
+    try:
+        for item in target.iterdir():
+            if item.is_file():
+                size_bytes = item.stat().st_size
+                total_size += size_bytes
+                file_count += 1
+                entries.append({
+                    "name": item.name,
+                    "size": size_bytes,
+                    "size_str": _size_str(size_bytes),
+                })
+            elif item.is_dir():
+                dir_size = sum(
+                    f.stat().st_size for f in item.rglob("*") if f.is_file()
+                ) if any(item.iterdir()) else 0
+                total_size += dir_size
+                file_count += sum(1 for _ in item.rglob("*") if _.is_file())
+                entries.append({
+                    "name": item.name + "/",
+                    "size": dir_size,
+                    "size_str": _size_str(dir_size),
+                })
+    except PermissionError as e:
+        return CommandResult(success=False, error=f"权限不足: {e}")
+
+    entries.sort(key=lambda x: x["size"], reverse=True)
+    top_entries = entries[:top]
+
+    return CommandResult(success=True, data={
+        "entries": top_entries,
+        "total_size": _size_str(total_size),
+        "file_count": file_count,
+        "path": str(target),
+    })
+
+
+# ── cleanup：缓存清理 ─────────────────────────────
+
+# 常见缓存目录（相对 ~ 或绝对路径）
+CACHE_PATTERNS: list[str] = [
+    "*/__pycache__",
+    "*/.pytest_cache",
+    "*/node_modules/.cache",
+    "*/.mypy_cache",
+    "*/.ruff_cache",
+    "*/*.pyc",
+    "*/.DS_Store",
+    "*/Thumbs.db",
+]
+
+
+def cleanup(path: str, dry_run: bool = False) -> CommandResult:
+    """清理缓存文件和临时文件"""
+    target = Path(path).expanduser().resolve()
+    if not target.is_dir():
+        return CommandResult(success=False, error=f"目录不存在: {path}")
+
+    stats = {"已清理": 0, "已释放": 0, "跳过": 0}
+    details: list[dict[str, Any]] = []
+
+    # 清理 __pycache__
+    for pycache in target.rglob("__pycache__"):
+        if pycache.is_dir():
+            size = sum(f.stat().st_size for f in pycache.rglob("*") if f.is_file())
+            if not dry_run:
+                import shutil
+                shutil.rmtree(pycache, ignore_errors=True)
+            stats["已清理"] += 1
+            stats["已释放"] += size
+            details.append({"name": str(pycache.relative_to(target)), "size": _size_str(size)})
+
+    # 清理 .pyc 文件
+    for pyc in target.rglob("*.pyc"):
+        if pyc.is_file():
+            size = pyc.stat().st_size
+            if not dry_run:
+                pyc.unlink()
+            stats["已清理"] += 1
+            stats["已释放"] += size
+            details.append({"name": str(pyc.relative_to(target)), "size": _size_str(size)})
+
+    # 清理 .DS_Store
+    for ds in target.rglob(".DS_Store"):
+        if ds.is_file():
+            size = ds.stat().st_size
+            if not dry_run:
+                ds.unlink()
+            stats["已清理"] += 1
+            stats["已释放"] += size
+            details.append({"name": str(ds.relative_to(target)), "size": _size_str(size)})
+
+    msg = f"清理完成: 清理了 {stats['已清理']} 个文件，释放 {_size_str(stats['已释放'])}"
+    if dry_run:
+        msg = f"[预览] 将清理 {stats['已清理']} 个文件，释放 {_size_str(stats['已释放'])}"
+
+    return CommandResult(success=True, data={
+        "stats": stats,
+        "details": details,
+        "message": msg,
+        "path": str(target),
+    })