File & Directory Automation

from pathlib import Path
import tempfile
import os
 
def atomic_write(path: Path, data: str) -> None:
    path.parent.mkdir(parents=True, exist_ok=True)
    fd, tmp = tempfile.mkstemp(dir=path.parent, text=True)
    try:
        with os.fdopen(fd, "w", encoding="utf-8") as f:
            f.write(data)
        os.replace(tmp, path)
    except Exception:
        os.unlink(tmp)
        raise

from __future__ import annotations
 
import json
import os
import tempfile
from dataclasses import dataclass
from pathlib import Path
 
@dataclass(frozen=True)
class FileRecord:
    path: Path
    size: int
 
def discover_csv(root: Path) -> list[FileRecord]:
    records: list[FileRecord] = []
    for path in root.glob("**/*.csv"):
        if path.is_file():
            size = path.stat().st_size
            if size == 0:
                raise ValueError(f"empty csv: {path}")
            records.append(FileRecord(path, size))
    return records
 
def atomic_write_json(path: Path, payload: dict) -> None:
    path.parent.mkdir(parents=True, exist_ok=True)
    fd, tmp = tempfile.mkstemp(dir=path.parent, suffix=".tmp")
    try:
        with os.fdopen(fd, "w", encoding="utf-8") as f:
            json.dump(payload, f, indent=2)
        os.replace(tmp, path)
    except Exception:
        os.unlink(tmp)
        raise
 
def main(root: Path, manifest: Path) -> int:
    files = discover_csv(root)
    atomic_write_json(
        manifest,
        {"count": len(files), "files": [{"path": str(r.path), "size": r.size} for r in files]},
    )
    return 0
 
if __name__ == "__main__":
    raise SystemExit(main(Path("data/incoming"), Path("data/manifest.json")))

import shutil
from pathlib import Path
 
shutil.copy2(Path("src.txt"), Path("backup/src.txt"))  # preserves metadata
shutil.rmtree(Path("tmp/build"), ignore_errors=False)