Spaces:
Running
Running
| from __future__ import annotations | |
| from datetime import datetime | |
| from uuid import uuid4 | |
| from sqlalchemy import ( | |
| JSON, | |
| CheckConstraint, | |
| DateTime, | |
| ForeignKey, | |
| Index, | |
| Integer, | |
| String, | |
| Text, | |
| UniqueConstraint, | |
| ) | |
| from sqlalchemy.orm import Mapped, mapped_column | |
| from app.security.models import Base, utcnow | |
| from app.projects.models.project import Project, ProjectGenerationJob | |
| class ProjectEditorState(Base): | |
| """The one authoritative current editor document for a project.""" | |
| __tablename__ = "project_editor_states" | |
| __table_args__ = ( | |
| UniqueConstraint("project_id", name="uq_project_editor_state_project"), | |
| Index("ix_project_editor_states_workspace_project", "workspace_id", "project_id"), | |
| ) | |
| id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4())) | |
| workspace_id: Mapped[str] = mapped_column( | |
| String(36), ForeignKey("workspaces.id", ondelete="RESTRICT"), nullable=False | |
| ) | |
| project_id: Mapped[str] = mapped_column( | |
| String(36), ForeignKey("projects.id", ondelete="RESTRICT"), nullable=False | |
| ) | |
| revision: Mapped[int] = mapped_column(Integer, nullable=False, default=1) | |
| schema_version: Mapped[int] = mapped_column(Integer, nullable=False) | |
| state_json: Mapped[dict[str, object]] = mapped_column("state", JSON, nullable=False, default=dict) | |
| created_at: Mapped[datetime] = mapped_column( | |
| DateTime(timezone=True), nullable=False, default=utcnow | |
| ) | |
| updated_at: Mapped[datetime] = mapped_column( | |
| DateTime(timezone=True), nullable=False, default=utcnow, onupdate=utcnow | |
| ) | |
| updated_by: Mapped[str] = mapped_column( | |
| String(36), ForeignKey("users.id", ondelete="RESTRICT"), nullable=False | |
| ) | |
| class ProjectRenderJob(Base): | |
| """A durable render of one immutable editor revision snapshot.""" | |
| __tablename__ = "project_render_jobs" | |
| __table_args__ = ( | |
| UniqueConstraint( | |
| "project_id", "editor_revision", "idempotency_key", | |
| name="uq_project_render_idempotency", | |
| ), | |
| CheckConstraint( | |
| "status in ('queued', 'processing', 'completed', 'failed', 'cancelling', 'cancelled')", | |
| name="ck_project_render_status", | |
| ), | |
| Index("ix_project_render_jobs_workspace_status", "workspace_id", "status"), | |
| Index("ix_project_render_jobs_project_created", "project_id", "created_at"), | |
| Index("ix_project_render_jobs_dispatch", "status", "next_attempt_at"), | |
| ) | |
| id: Mapped[str] = mapped_column(String(36), primary_key=True, default=lambda: str(uuid4())) | |
| workspace_id: Mapped[str] = mapped_column( | |
| String(36), ForeignKey("workspaces.id", ondelete="RESTRICT"), nullable=False | |
| ) | |
| project_id: Mapped[str] = mapped_column( | |
| String(36), ForeignKey("projects.id", ondelete="RESTRICT"), nullable=False | |
| ) | |
| editor_revision: Mapped[int] = mapped_column(Integer, nullable=False) | |
| editor_schema_version: Mapped[int] = mapped_column(Integer, nullable=False) | |
| editor_state_json: Mapped[dict[str, object]] = mapped_column("editor_state", JSON, nullable=False) | |
| render_settings_json: Mapped[dict[str, object]] = mapped_column("render_settings", JSON, nullable=False) | |
| request_fingerprint: Mapped[str] = mapped_column(String(64), nullable=False) | |
| idempotency_key: Mapped[str] = mapped_column(String(255), nullable=False) | |
| requested_by: Mapped[str] = mapped_column( | |
| String(36), ForeignKey("users.id", ondelete="RESTRICT"), nullable=False | |
| ) | |
| status: Mapped[str] = mapped_column(String(32), nullable=False, default="queued") | |
| attempt_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0) | |
| max_attempts: Mapped[int] = mapped_column(Integer, nullable=False, default=3) | |
| next_attempt_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True)) | |
| output_asset_id: Mapped[str | None] = mapped_column( | |
| String(36), ForeignKey("media_assets.id", ondelete="RESTRICT") | |
| ) | |
| error_code: Mapped[str | None] = mapped_column(String(100)) | |
| error_message: Mapped[str | None] = mapped_column(Text) | |
| created_at: Mapped[datetime] = mapped_column( | |
| DateTime(timezone=True), nullable=False, default=utcnow | |
| ) | |
| started_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True)) | |
| completed_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True)) | |
| cancelled_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True)) | |
| updated_at: Mapped[datetime] = mapped_column( | |
| DateTime(timezone=True), nullable=False, default=utcnow, onupdate=utcnow | |
| ) | |
| from app.projects.models.collaboration import ( # noqa: E402 | |
| ApprovalRequest, | |
| ApprovalWorkflow, | |
| CollaborationActivity, | |
| Invitation, | |
| ProjectCollaborator, | |
| ReviewComment, | |
| Team, | |
| TeamMember, | |
| ) | |
| from app.projects.models.notifications import NotificationPreference # noqa: E402 | |
| __all__ = [ | |
| "ApprovalRequest", | |
| "ApprovalWorkflow", | |
| "CollaborationActivity", | |
| "Invitation", | |
| "NotificationPreference", | |
| "Project", | |
| "ProjectCollaborator", | |
| "ProjectEditorState", | |
| "ProjectGenerationJob", | |
| "ProjectRenderJob", | |
| "ReviewComment", | |
| "Team", | |
| "TeamMember", | |
| ] | |