Diff summary for Restic backups

restic-diff-summary.py

#!/usr/bin/env python3
import sys
from collections import defaultdict
import os

# Cores ANSI
GREEN = "\033[32m"
RED = "\033[31m"
YELLOW = "\033[33m"
BOLD = "\033[1m"
RESET = "\033[0m"

changes = defaultdict(lambda: {"added": 0, "removed": 0, "modified": 0})

for line in sys.stdin:
    line = line.strip()
    if not line:
        continue

    status = line[0]
    path = line[1:].strip()
    directory = os.path.dirname(path) or "/"

    if status == "+":
        changes[directory]["added"] += 1
    elif status == "-":
        changes[directory]["removed"] += 1
    elif status.upper() == "M":
        changes[directory]["modified"] += 1
    else:
        changes[directory]["modified"] += 1

for directory, stats in sorted(changes.items()):
    parts = []

    if stats["added"] > 0:
        parts.append(f"{GREEN}+{stats['added']}{RESET}")

    if stats["removed"] > 0:
        parts.append(f"{RED}-{stats['removed']}{RESET}")

    if stats["modified"] > 0:
        parts.append(f"{YELLOW}M{stats['modified']}{RESET}")

    if not parts:
        continue

    total = stats["added"] + stats["removed"] + stats["modified"]

    # Se passou de 100 mudanças, deixa a linha inteira em negrito
    if total > 100:
        print(f"{BOLD}{directory}: " + "  ".join(parts) + RESET)
    else:
        print(f"{directory}: " + "  ".join(parts))