Certify, sell, and verify synthetic datasets

from pathlib import Path
from concurrent.futures import ThreadPoolExecutor
import hashlib, requests, time

def certify_dataset(path: Path) -> dict:
    sha = hashlib.sha256(path.read_bytes()).hexdigest()
    cert = requests.post(
        "https://certifieddata.io/api/certify",
        headers={"Authorization": f"Bearer {CD_API_KEY}"},
        json={
            "artifact_type": "synthetic_dataset",
            "sha256": sha,
            "metadata": { "filename": path.name }
        }).json()
    return {
        "file": str(path),
        "sha256": sha,
        "certificate_id": cert["certificate_id"]
    }

files = list(Path("./datasets").glob("*.parquet"))

# Certify 100 files with 5 parallel workers
with ThreadPoolExecutor(max_workers=5) as exe:
    catalog = list(exe.map(certify_dataset, files))

# catalog is now: [{file, sha256, certificate_id}, ...]

from certifieddata_payments import CertifiedDataPayments

def sell_dataset(certificate_id: str,
                 buyer_api_key: str) -> dict:
    """Buyer agent hits this endpoint to purchase."""
    item = catalog[certificate_id]
    cdp  = CertifiedDataPayments(api_key=buyer_api_key)

    tx = cdp.transactions.create({
        "amount":   item["price_cents"],
        "currency": "usd",
        "payee_id": "merch_dataset_seller",
        "rail":     "stripe",
    })

    cdp.transactions.attach_links(
        tx["transaction_id"], {
            "certificate_id": certificate_id,
            "artifact_hash":  f"sha256:{item['sha256']}",
            "decision_record_id": f"dec_{tx['transaction_id']}",
        }
    )

    capture = cdp.transactions.capture(
        tx["transaction_id"]
    )
    receipt = capture["receipt"]

    return {
        "receipt":      receipt,
        "cert_url":     item["certificate_url"],
        "verify_url":   f"certifieddata.io/api/payments"
                        f"/verify/{receipt['receipt_id']}",
    }