"""
MongoDB document models and helper functions.
"""

from datetime import datetime, timezone
from typing import Optional


def create_inspection_document(
    inspection_id: str,
    url: str,
    status: str,
    overall_score: int,
    grade: str,
    categories: dict,
    summary: dict,
    created_at: datetime,
    completed_at: Optional[datetime] = None,
    duration_seconds: Optional[float] = None,
) -> dict:
    """Create a MongoDB document for the inspections collection."""
    return {
        "inspection_id": inspection_id,
        "url": url,
        "status": status,
        "created_at": created_at,
        "completed_at": completed_at,
        "duration_seconds": duration_seconds,
        "overall_score": overall_score,
        "grade": grade,
        "categories": categories,
        "summary": summary,
    }