Spaces:
Sleeping
Sleeping
File size: 3,326 Bytes
1b2323a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 | 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
class Project(Base):
"""A durable workspace-owned container for future creative resources."""
__tablename__ = "projects"
__table_args__ = (
CheckConstraint("status in ('active', 'archived')", name="ck_projects_status"),
Index("ix_projects_workspace", "workspace_id"),
Index("ix_projects_workspace_status", "workspace_id", "status"),
Index("ix_projects_workspace_updated", "workspace_id", "updated_at"),
Index("ix_projects_created_by", "created_by"),
)
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
)
created_by: Mapped[str] = mapped_column(
String(36), ForeignKey("users.id", ondelete="RESTRICT"), nullable=False
)
name: Mapped[str] = mapped_column(String(200), nullable=False)
description: Mapped[str | None] = mapped_column(Text)
status: Mapped[str] = mapped_column(String(16), nullable=False, default="active")
thumbnail_asset_id: Mapped[str | None] = mapped_column(
String(36), ForeignKey("media_assets.id", ondelete="SET NULL")
)
brand_kit_id: Mapped[str | None] = mapped_column(
String(36), ForeignKey("brand_kits.id", ondelete="RESTRICT")
)
metadata_json: Mapped[dict[str, object]] = mapped_column(
"metadata", 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)
archived_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
class ProjectGenerationJob(Base):
"""Associates the existing durable generation job with one project."""
__tablename__ = "project_generation_jobs"
__table_args__ = (
UniqueConstraint("generation_job_id", name="uq_project_generation_job"),
Index(
"ix_project_generation_jobs_workspace_project_created",
"workspace_id",
"project_id",
"created_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
)
generation_job_id: Mapped[str] = mapped_column(
String(36), ForeignKey("generation_jobs.id", ondelete="RESTRICT"), nullable=False
)
attached_by: Mapped[str] = mapped_column(
String(36), ForeignKey("users.id", ondelete="RESTRICT"), nullable=False
)
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, default=utcnow)
|