"""Definition of Job DTO."""

from uuid import uuid4, UUID
from pydantic import Field
from PIL import Image

from .type_checking_base_model import TypeCheckingBaseModel
from .annotation import Annotation


class Job(TypeCheckingBaseModel):
    """Data Transfer Object representing a Job."""

    class Config:
        """Pydantic configuration for Job DTO."""

        arbitrary_types_allowed = True

    job_id: UUID = Field(default_factory=uuid4)
    image: Image.Image
    annotation: Annotation