import uuid
from datetime import datetime

from pydantic import BaseModel

from app.models.export_file import ExportFormat


class ExportRequest(BaseModel):
    format: ExportFormat


class ExportFileResponse(BaseModel):
    id: uuid.UUID
    protocol_id: uuid.UUID
    format: ExportFormat
    file_size: int
    created_at: datetime

    model_config = {"from_attributes": True}