from pydantic import BaseModel, Field


class TranslationResponse(BaseModel):
    document_id: str = Field(alias="documentId")
    failure_reason: str | None = Field(alias="failureReason", default=None)
    href: str
    id: str
    name: str
    request_element_id: str = Field(alias="requestElementId")
    request_state: str = Field(alias="requestState")
    result_document_id: str = Field(alias="resultDocumentId")
    result_element_ids: list[str] | None = Field(alias="resultElementIds", default=None)
    result_external_data_ids: list[str] | None = Field(
        alias="resultExternalDataIds", default=None
    )
    result_workspace_id: str | None = Field(alias="resultWorkspaceId", default=None)
    version_id: str | None = Field(alias="versionId", default=None)
    view_ref: str | None = Field(alias="viewRef", default=None)
    workspace_id: str = Field(alias="workspaceId")


class TranslationStatusResponse(BaseModel):
    id: str
    request_state: str = Field(alias="requestState")
    request_element_id: str = Field(alias="requestElementId")
    result_external_data_ids: list[str] | None = Field(
        alias="resultExternalDataIds", default=None
    )
    export_rule_file_name: str | None = Field(alias="exportRuleFileName", default=None)
    version_id: str | None = Field(alias="versionId", default=None)
    workspace_id: str = Field(alias="workspaceId")
    document_id: str = Field(alias="documentId")
    result_element_ids: list[str] | None = Field(alias="resultElementIds", default=None)
    result_document_id: str = Field(alias="resultDocumentId")
    failure_reason: str | None = Field(alias="failureReason", default=None)
    result_workspace_id: str | None = Field(alias="resultWorkspaceId", default=None)
    name: str
    href: str