basicmachines-co-basic-memory/create_csv_exhibits.py

#!/usr/bin/env python3
"""
Create CSV exhibits for individual contributors
"""

import json
import csv
from pathlib import Path


def create_csv_exhibits():
    """Create CSV Exhibit A files for each contributor."""

    # Read the JSON inventory
    inventory_files = list(Path("legal_inventory_main").glob("*.json"))
    if not inventory_files:
        print("Error: No JSON inventory files found")
        return

    inventory_file = inventory_files[0]  # Use the most recent one
    print(f"Using inventory file: {inventory_file}")

    with open(inventory_file, "r") as f:
        data = json.load(f)

    files = data["files"]

    # Create output directory
    output_dir = Path("legal_exhibits")
    output_dir.mkdir(exist_ok=True)

    # Contributors we need exhibits for
    target_contributors = {"jope-bm": "joe_exhibit_a.csv", "Drew Cain": "drew_cain_exhibit_a.csv"}

    print("Creating CSV exhibits for contributors...")

    for contributor_key, filename in target_contributors.items():
        # Find files for this contributor
        contributor_files = []

        for file_info in files:
            # Check if this contributor is listed in the file's contributors
            for contrib in file_info.get("contributors", []):
                if contributor_key in contrib["name"]:
                    contributor_files.append(file_info)
                    break

        if not contributor_files:
            print(f"No files found for {contributor_key}")
            continue

        # Sort files by path
        contributor_files.sort(key=lambda x: x["path"])

        # Create CSV file
        csv_file = output_dir / filename

        with open(csv_file, "w", newline="", encoding="utf-8") as csvfile:
            fieldnames = [
                "file_path",
                "file_name",
                "category",
                "size_bytes",
                "modified_date",
                "primary_author",
                "all_contributors",
                "sha256_hash",
            ]

            writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
            writer.writeheader()

            for file_info in contributor_files:
                contributors_list = "; ".join([c["name"] for c in file_info["contributors"]])

                writer.writerow(
                    {
                        "file_path": file_info["path"],
                        "file_name": file_info["name"],
                        "category": file_info["category"],
                        "size_bytes": file_info["size_bytes"],
                        "modified_date": file_info["modified_time"][:10],
                        "primary_author": file_info["primary_author"],
                        "all_contributors": contributors_list,
                        "sha256_hash": file_info["sha256_hash"],
                    }
                )

        print(f"Created CSV exhibit for {contributor_key}: {csv_file}")
        print(f"  - {len(contributor_files)} files")
        print(f"  - {sum(f['size_bytes'] for f in contributor_files):,} bytes")


if __name__ == "__main__":
    create_csv_exhibits()