convert_notebooks.py

#!/usr/bin/env python3
from __future__ import annotations

import argparse
from pathlib import Path

import nbformat
from nbconvert import WebPDFExporter


def ipynb_to_pdf(ipynb_path: Path, pdf_path: Path):
    try:
        with open(ipynb_path, encoding="utf-8") as f:
            nb = nbformat.read(f, as_version=4)

        pdf_exporter = WebPDFExporter()
        pdf_exporter.exclude_input_prompt = True
        pdf_exporter.exclude_output_prompt = True

        pdf_data, _ = pdf_exporter.from_notebook_node(nb)

        with open(pdf_path, "wb") as f:
            f.write(pdf_data)

        print(f"✅ Converted: {ipynb_path.name} -> {pdf_path.name}")
    except Exception as e:
        print(f"❌ Failed to convert {ipynb_path.name} to pdf: {e}")


def sync_notebooks_to_pdf(directory_path: str, output_path: str, force: bool = False):
    base_path = Path(directory_path)

    if output_path:
        out_base = Path(output_path)
    else:
        out_base = base_path / "resources"

    out_base.mkdir(parents=True, exist_ok=True)

    notebooks = list(base_path.glob("*.ipynb"))

    if not notebooks:
        print(f"⚠️ No notebooks found in: {directory_path}")
        return

    for ipynb_path in notebooks:
        pdf_path = out_base / ipynb_path.with_suffix(".pdf").name
        should_convert = False

        if force:
            should_convert = True
        elif not pdf_path.exists():
            should_convert = True
        else:
            ipynb_mtime = ipynb_path.stat().st_mtime
            pdf_mtime = pdf_path.stat().st_mtime

            if ipynb_mtime > pdf_mtime:
                should_convert = True

        if should_convert:
            ipynb_to_pdf(ipynb_path, pdf_path)
        else:
            print(f"ℹ️ Skipped: {ipynb_path.name} (PDF is up to date)")


def main():
    parser = argparse.ArgumentParser(
        description="Sync jupyter notebooks to PDF with math support",
    )

    parser.add_argument(
        "--notebooks_dir",
        "-d",
        required=True,
        help="Path to directory with notebooks",
    )

    parser.add_argument(
        "--output_dir",
        "-o",
        required=False,
        default=None,
        help="Output directory for PDF files",
    )

    parser.add_argument(
        "--force",
        "-f",
        action="store_true",
        help="Force re-export of all notebooks even if they haven't changed",
    )

    args = parser.parse_args()

    sync_notebooks_to_pdf(args.notebooks_dir, args.output_dir, args.force)


if __name__ == "__main__":
    main()