Daniil
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from __future__ import annotations
|
|
|
|
import uuid
|
|
|
|
from sqlalchemy import JSON, ForeignKey, String
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from cpv3.db.base import Base, BaseModelMixin
|
|
|
|
|
|
class Transcription(Base, BaseModelMixin):
|
|
__tablename__ = "transcriptions"
|
|
|
|
project_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
ForeignKey("projects.id", ondelete="RESTRICT"),
|
|
nullable=True,
|
|
index=True,
|
|
)
|
|
source_file_id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True), ForeignKey("files.id", ondelete="RESTRICT"), index=True
|
|
)
|
|
artifact_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
ForeignKey("artifact_media_files.id", ondelete="RESTRICT"),
|
|
nullable=True,
|
|
index=True,
|
|
)
|
|
|
|
engine: Mapped[str] = mapped_column(String(32), default="LOCAL_WHISPER")
|
|
language: Mapped[str | None] = mapped_column(String(3), nullable=True)
|
|
|
|
document: Mapped[dict] = mapped_column(JSON)
|
|
transcribe_options: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|