Daniil
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import uuid
|
|
|
|
from sqlalchemy import Boolean, Float, ForeignKey, Integer, JSON, String, Text
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from cpv3.db.base import Base, BaseModelMixin
|
|
|
|
|
|
class MediaFile(Base, BaseModelMixin):
|
|
__tablename__ = "media_files"
|
|
|
|
owner_id: Mapped[uuid.UUID] = mapped_column(
|
|
UUID(as_uuid=True), ForeignKey("users.id", ondelete="RESTRICT"), index=True
|
|
)
|
|
project_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
ForeignKey("projects.id", ondelete="RESTRICT"),
|
|
nullable=True,
|
|
index=True,
|
|
)
|
|
|
|
duration_seconds: Mapped[float] = mapped_column(Float)
|
|
frame_rate: Mapped[float | None] = mapped_column(Float, nullable=True)
|
|
width: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
height: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
|
|
|
probe_json: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
notes: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
meta: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
|
|
is_deleted: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
|
|
|
|
class ArtifactMediaFile(Base, BaseModelMixin):
|
|
__tablename__ = "artifact_media_files"
|
|
|
|
project_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
ForeignKey("projects.id", ondelete="RESTRICT"),
|
|
nullable=True,
|
|
index=True,
|
|
)
|
|
file_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True), ForeignKey("files.id", ondelete="RESTRICT"), nullable=True, index=True
|
|
)
|
|
media_file_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
UUID(as_uuid=True),
|
|
ForeignKey("media_files.id", ondelete="RESTRICT"),
|
|
nullable=True,
|
|
index=True,
|
|
)
|
|
|
|
artifact_type: Mapped[str] = mapped_column(String(32), default="TRANSCRIPTION_JSON")
|
|
is_deleted: Mapped[bool] = mapped_column(Boolean, default=False)
|