mirror of
https://github.com/bellingcat/whisperbox-transcribe.git
synced 2026-06-12 05:28:34 +03:00
50 lines
879 B
Python
50 lines
879 B
Python
import enum
|
|
from datetime import datetime
|
|
from typing import Any, Optional
|
|
from uuid import UUID
|
|
|
|
from pydantic import AnyHttpUrl, BaseModel
|
|
|
|
|
|
class WithDbFields(BaseModel):
|
|
id: UUID
|
|
created_at: datetime
|
|
updated_at: Optional[datetime]
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
|
|
class ArtifactType(str, enum.Enum):
|
|
raw_transcript = "raw_transcript"
|
|
|
|
|
|
class JobType(str, enum.Enum):
|
|
transcript = "transcript"
|
|
|
|
|
|
class JobStatus(str, enum.Enum):
|
|
create = "create"
|
|
processing = "processing"
|
|
error = "error"
|
|
success = "success"
|
|
|
|
|
|
class JobMeta(BaseModel):
|
|
language: Optional[str]
|
|
task_id: Optional[UUID]
|
|
|
|
|
|
class Job(WithDbFields):
|
|
status: JobStatus
|
|
type: JobType
|
|
url: AnyHttpUrl
|
|
meta: Optional[JobMeta]
|
|
|
|
|
|
class Artifact(WithDbFields):
|
|
# TODO: narrow type
|
|
data: Optional[Any]
|
|
job_id: UUID
|
|
type: ArtifactType
|