TieMeasureFlow/src/backend/api/routers/reports.py

"""Reports router — PDF report generation endpoints for Metrologist."""
from datetime import datetime

from fastapi import APIRouter, Depends, HTTPException, Query, status
from fastapi.responses import Response
from sqlalchemy.ext.asyncio import AsyncSession

from src.backend.database import get_db
from src.backend.api.middleware.api_key import require_metrologist
from src.backend.models.orm.user import User
from src.backend.services.report_service import generate_measurement_report, generate_spc_report

router = APIRouter(prefix="/api/reports", tags=["reports"])


@router.get("/spc")
async def download_spc_report(
    recipe_id: int = Query(...),
    subtask_id: int = Query(...),
    version_id: int | None = Query(None),
    date_from: datetime | None = Query(None),
    date_to: datetime | None = Query(None),
    operator_id: int | None = Query(None),
    lot_number: str | None = Query(None),
    serial_number: str | None = Query(None),
    user: User = Depends(require_metrologist),
    db: AsyncSession = Depends(get_db),
) -> Response:
    """Generate and download SPC PDF report."""
    try:
        pdf_bytes = await generate_spc_report(
            db, recipe_id, subtask_id,
            version_id, date_from, date_to,
            operator_id, lot_number, serial_number,
        )
    except Exception as e:
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"Report generation failed: {str(e)}",
        )

    filename = f"spc_report_recipe{recipe_id}_st{subtask_id}.pdf"
    return Response(
        content=pdf_bytes,
        media_type="application/pdf",
        headers={"Content-Disposition": f'attachment; filename="{filename}"'},
    )


@router.get("/measurements")
async def download_measurement_report(
    recipe_id: int = Query(...),
    subtask_id: int | None = Query(None),
    version_id: int | None = Query(None),
    date_from: datetime | None = Query(None),
    date_to: datetime | None = Query(None),
    operator_id: int | None = Query(None),
    lot_number: str | None = Query(None),
    serial_number: str | None = Query(None),
    user: User = Depends(require_metrologist),
    db: AsyncSession = Depends(get_db),
) -> Response:
    """Generate and download measurement table PDF report."""
    try:
        pdf_bytes = await generate_measurement_report(
            db, recipe_id, subtask_id,
            version_id, date_from, date_to,
            operator_id, lot_number, serial_number,
        )
    except Exception as e:
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"Report generation failed: {str(e)}",
        )

    filename = f"measurements_report_recipe{recipe_id}.pdf"
    return Response(
        content=pdf_bytes,
        media_type="application/pdf",
        headers={"Content-Disposition": f'attachment; filename="{filename}"'},
    )