Source code for plom.create.build_scrap_paper_with_qrcodes

#!/usr/bin/env python3

# SPDX-License-Identifier: AGPL-3.0-or-later
# Copyright (C) 2023 Andrew Rechnitzer
# Copyright (C) 2023-2026 Colin B. Macdonald

import shutil
import subprocess
import tempfile
from importlib import resources
from pathlib import Path

import segno

from plom.common.tpv_utils import encodeScrapPaperCode
import plom.create




def build_scrap_paper_pdf(destination_dir=None, *, latex_papersize: str = "") -> Path:
    """Build the scrap paper pdf file.

    Similar to :func:`build_extra_page_pdf`.
    """
    if destination_dir is None:
        destination_dir = Path.cwd()

    src_tex = (resources.files(plom.create) / "scrap_paper_src.tex").read_text()
    if latex_papersize:
        src_tex = src_tex.replace("letterpaper", latex_papersize)
    with tempfile.TemporaryDirectory() as tmpdirname:
        tmp_path = Path(tmpdirname)
        with open(tmp_path / "scrap_paper.tex", "w") as fh:
            fh.write(src_tex)

        for crn in range(1, 9):
            qr = segno.make_micro(encodeScrapPaperCode(crn))
            # MyPy complains about pathlib.Path here but it works
            qr.save(tmp_path / f"qr_crn_{crn}.png", border=2, scale=4)  # type: ignore[arg-type]

        subprocess.run(
            (
                "latexmk",
                "-pdf",
                "-interaction=nonstopmode",
                "-no-shell-escape",
                "scrap_paper",
            ),
            cwd=tmp_path,
            stdout=subprocess.DEVNULL,
        )
        shutil.copy(tmp_path / "scrap_paper.pdf", destination_dir)
    return Path(destination_dir) / "scrap_paper.pdf"



if __name__ == "__main__":
    build_scrap_paper_pdf()