"""TranscodeJob Schemas - GENERATED FILE"""

from datetime import datetime
from enum import Enum
from typing import Any, Dict, List, Optional
from uuid import UUID

from .base import BaseSchema


class JobStatus(str, Enum):
    PENDING = "pending"
    PROCESSING = "processing"
    COMPLETED = "completed"
    FAILED = "failed"
    CANCELLED = "cancelled"


class JobCreate(BaseSchema):
    """JobCreate schema."""
    source_asset_id: UUID
    preset_id: Optional[UUID] = None
    preset_snapshot: Dict[str, Any]
    trim_start: Optional[float] = None
    trim_end: Optional[float] = None
    output_filename: str = ""
    output_path: Optional[str] = None
    output_asset_id: Optional[UUID] = None
    progress: float = 0.0
    current_frame: Optional[int] = None
    current_time: Optional[float] = None
    speed: Optional[str] = None
    celery_task_id: Optional[str] = None
    execution_arn: Optional[str] = None
    priority: int = 0
    started_at: Optional[datetime] = None
    completed_at: Optional[datetime] = None

class JobUpdate(BaseSchema):
    """JobUpdate schema."""
    source_asset_id: Optional[UUID] = None
    preset_id: Optional[UUID] = None
    preset_snapshot: Optional[Dict[str, Any]] = None
    trim_start: Optional[float] = None
    trim_end: Optional[float] = None
    output_filename: Optional[str] = None
    output_path: Optional[str] = None
    output_asset_id: Optional[UUID] = None
    status: Optional[JobStatus] = None
    progress: Optional[float] = None
    current_frame: Optional[int] = None
    current_time: Optional[float] = None
    speed: Optional[str] = None
    error_message: Optional[str] = None
    celery_task_id: Optional[str] = None
    execution_arn: Optional[str] = None
    priority: Optional[int] = None
    started_at: Optional[datetime] = None
    completed_at: Optional[datetime] = None

class JobResponse(BaseSchema):
    """JobResponse schema."""
    id: UUID
    source_asset_id: UUID
    preset_id: Optional[UUID] = None
    preset_snapshot: Dict[str, Any]
    trim_start: Optional[float] = None
    trim_end: Optional[float] = None
    output_filename: str = ""
    output_path: Optional[str] = None
    output_asset_id: Optional[UUID] = None
    status: JobStatus = "JobStatus.PENDING"
    progress: float = 0.0
    current_frame: Optional[int] = None
    current_time: Optional[float] = None
    speed: Optional[str] = None
    error_message: Optional[str] = None
    celery_task_id: Optional[str] = None
    execution_arn: Optional[str] = None
    priority: int = 0
    created_at: Optional[datetime] = None
    started_at: Optional[datetime] = None
    completed_at: Optional[datetime] = None