clean_memory.py

#!/usr/bin/env python3
"""
Project Junk Cleaner

Safely removes Python cache files, build artifacts, and temporary files.

Example:
    python clean.py
    python clean.py --dry-run
    python clean.py --yes
"""

from __future__ import annotations

import argparse
import shutil
from pathlib import Path

# ---------------------------------------------------------------------
# Configuration
# ---------------------------------------------------------------------

PROTECTED_DIRS = {
    ".git",
    "venv",
    ".venv",
    "env",
    ".idea",
}

JUNK_DIRS = {
    "__pycache__",
    ".pytest_cache",
    ".mypy_cache",
    ".ruff_cache",
    ".tox",
    ".coverage",
    ".hypothesis",
    "build",
    "dist",
    "*.egg-info",
}

JUNK_FILES = {
    "*.pyc",
    "*.pyo",
    "*.pyd",
    "*.tmp",
    "*.log",
    "*.bak",
    "*.swp",
    "*.swo",
    ".DS_Store",
    "Thumbs.db",
    ".coverage",
}

# ---------------------------------------------------------------------
# Helpers
# ---------------------------------------------------------------------


def find_junk(root: Path):
    directories = []
    files = []

    for path in root.rglob("*"):

        if path.is_symlink():
            continue

        if any(parent.name in PROTECTED_DIRS for parent in path.parents):
            continue

        if path.is_dir():
            for pattern in JUNK_DIRS:
                if path.match(pattern):
                    directories.append(path)
                    break

        elif path.is_file():
            for pattern in JUNK_FILES:
                if path.match(pattern):
                    files.append(path)
                    break

    return directories, files


def delete_directory(path: Path):
    shutil.rmtree(path, ignore_errors=True)


def delete_file(path: Path):
    path.unlink(missing_ok=True)


# ---------------------------------------------------------------------
# Main
# ---------------------------------------------------------------------


def cleaner():

    parser = argparse.ArgumentParser(
        description="Clean Python project junk files."
    )

    parser.add_argument(
        "--path",
        type=Path,
        default=Path.cwd(),
        help="Project root (default=current directory)",
    )

    parser.add_argument(
        "--dry-run",
        action="store_true",
        help="Show what would be removed.",
    )

    parser.add_argument(
        "--yes",
        action="store_true",
        help="Skip confirmation.",
    )

    args = parser.parse_args()

    root = args.path.resolve()

    dirs, files = find_junk(root)

    print(f"\nProject : {root}")
    print(f"Directories : {len(dirs)}")
    print(f"Files       : {len(files)}\n")

    if not dirs and not files:
        print("Nothing to clean.")
        return

    for d in dirs:
        print("[DIR ]", d)

    for f in files:
        print("[FILE]", f)

    if args.dry_run:
        return



    removed_dirs = 0
    removed_files = 0

    for d in dirs:
        delete_directory(d)
        removed_dirs += 1

    for f in files:
        delete_file(f)
        removed_files += 1

    print("\nCleanup complete.")
    print(f"Directories removed : {removed_dirs}")
    print(f"Files removed       : {removed_files}")