repro-bundle/scripts/replace_logbook_figure.py

#!/usr/bin/env python3
"""Replace one Trackio HTML figure payload while preserving cell metadata/pinning."""

import argparse
from pathlib import Path


def main() -> None:
    ap = argparse.ArgumentParser()
    ap.add_argument("page", type=Path)
    ap.add_argument("cell_id")
    ap.add_argument("html", type=Path)
    args = ap.parse_args()
    text = args.page.read_text(encoding="utf-8")
    marker = f'"id": "{args.cell_id}"'
    marker_at = text.index(marker)
    fence_at = text.index("````html\n", marker_at)
    payload_at = fence_at + len("````html\n")
    end_at = text.index("\n````", payload_at)
    replacement = args.html.read_text(encoding="utf-8").rstrip()
    args.page.write_text(text[:payload_at] + replacement + text[end_at:], encoding="utf-8")
    print(f"replaced {args.cell_id} from {args.html}")


if __name__ == "__main__":
    main()