fill_pdf.py

"""
Bytedoc — Fill a PDF form

Fills AcroForm fields in a PDF template with provided data.
Requires: pip install requests
"""

import os
import sys

import requests

API_KEY = os.environ["BYTEDOC_API_KEY"]
BASE_URL = "https://api.bytedoc.dev/v1"


def fill_pdf(
    template_id: str,
    data: dict,
    flatten: bool = True,
    filename: str | None = None,
) -> dict:
    """Fill a PDF template and return the document metadata."""
    response = requests.post(
        f"{BASE_URL}/documents/fill",
        headers={"Authorization": f"Bearer {API_KEY}"},
        json={
            "template_id": template_id,
            "data": data,
            "options": {
                "flatten": flatten,
                "filename": filename,
            },
        },
    )

    if not response.ok:
        error = response.json()
        raise RuntimeError(
            f"Bytedoc error ({response.status_code}): {error['error']['message']}"
        )

    return response.json()


def download_document(download_url: str, output_path: str) -> None:
    """Download a generated document to a local file."""
    response = requests.get(
        download_url,
        headers={"Authorization": f"Bearer {API_KEY}"},
    )
    response.raise_for_status()

    with open(output_path, "wb") as f:
        f.write(response.content)


# ---------------------------------------------------------------------------
# Usage
# ---------------------------------------------------------------------------

if __name__ == "__main__":
    result = fill_pdf(
        template_id="tpl_abc123",
        data={
            "full_name": "John Doe",
            "email": "john@example.com",
            "agree_terms": True,
        },
        flatten=True,
        filename="filled-form.pdf",
    )

    print(f"Document ID : {result['id']}")
    print(f"Download URL: {result['download_url']}")
    print(f"Expires at  : {result['expires_at']}")

    # Optionally download the file
    download_document(result["download_url"], "filled-form.pdf")
    print("Saved to filled-form.pdf")