test_contract_renderer.py

#!/usr/bin/env python3
"""
Test script for the contract renderer functionality.
This script demonstrates how to use the contract renderer to generate HTML and PDF documents.
"""

import argparse
import asyncio
import sys
from pathlib import Path

from prozorro.render.render.contract import render_contract_html, render_contract_pdf
from prozorro.render.utils.api import get_template_name
from prozorro.render.utils.contract import (
    generate_contract_template_context,
    generate_contract_template_filename,
)

# Define directories
FILE_DIR = Path(__file__).parent
MEDIA_DIR = FILE_DIR / "media"

# Create data directory if it doesn't exist
MEDIA_DIR.mkdir(exist_ok=True)

# Add the src directory to the Python path
sys.path.insert(0, str(Path(__file__).parent / "src"))


async def async_main():
    """Test the contract renderer with sample data."""

    # Parse command line arguments
    parser = argparse.ArgumentParser(description="Test contract renderer")
    parser.add_argument("--url", help="URL to fetch contract data from")
    parser.add_argument("--template", help="Template name to use for rendering")
    args = parser.parse_args()

    # Generate context and fetch data if URL provided
    print("=" * 50)
    print("Preparing data...")
    print("=" * 50)
    try:
        context = await generate_contract_template_context(args.url)
        if "contract_url" in context:
            print(f"Successfully fetched contract data from: {context['contract_url']}")
        if "tender_url" in context:
            print(f"Successfully fetched tender data from: {context['tender_url']}")
    except ValueError as e:
        print(f"Error fetching data: {e}")
        return 1

    # Get template name
    try:
        template_name = get_template_name(template=args.template, contract_data=context["contract"])
        print(f"Using template: {template_name}")
    except ValueError as e:
        print(f"Error: {e}")
        return 1

    print("=" * 50)
    print("Testing Contract Renderer...")
    print("=" * 50)

    filename = generate_contract_template_filename(context["contract"].get("contractID"), template_name)

    try:
        # Test HTML rendering
        print("1. Generating HTML...")
        html_content = render_contract_html(context, template_name)

        # Save HTML to file
        html_path = MEDIA_DIR / f"{filename}.html"
        with open(html_path, "w", encoding="utf-8") as f:
            f.write(html_content)
        print(f"   ✓ HTML saved to {html_path}")

        # Test PDF rendering
        print("2. Generating PDF...")
        pdf_content = render_contract_pdf(context, template_name)

        # Save PDF to file
        pdf_path = MEDIA_DIR / f"{filename}.pdf"
        with open(pdf_path, "wb") as f:
            f.write(pdf_content)
        print(f"   ✓ PDF saved to {pdf_path}")

    except Exception as e:
        print(f"Error during testing: {e}")
        return 1

    return 0


def main():
    """Entry point for the script."""
    return asyncio.run(async_main())


if __name__ == "__main__":
    exit(main())