commit
This commit is contained in:
5
backend/app/models/__init__.py
Normal file
5
backend/app/models/__init__.py
Normal file
@@ -0,0 +1,5 @@
|
||||
|
||||
from .project import Project
|
||||
from .ingredient import Ingredient, AssetType
|
||||
from .scene import Scene
|
||||
from .shot import Shot
|
||||
29
backend/app/models/ingredient.py
Normal file
29
backend/app/models/ingredient.py
Normal file
@@ -0,0 +1,29 @@
|
||||
|
||||
from sqlalchemy import Column, String, DateTime, func, ForeignKey, Enum
|
||||
from sqlalchemy.dialects.postgresql import UUID, JSONB
|
||||
import uuid
|
||||
from sqlalchemy.orm import relationship
|
||||
import enum
|
||||
|
||||
from app.db.session import Base
|
||||
|
||||
class AssetType(str, enum.Enum):
|
||||
Character = "Character"
|
||||
Location = "Location"
|
||||
Object = "Object"
|
||||
Style = "Style"
|
||||
|
||||
class Ingredient(Base):
|
||||
__tablename__ = "ingredients"
|
||||
|
||||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||||
project_id = Column(UUID(as_uuid=True), ForeignKey("projects.id", ondelete="CASCADE"), nullable=False)
|
||||
name = Column(String, nullable=False)
|
||||
type = Column(Enum(AssetType, name="asset_type"), nullable=False)
|
||||
s3_key = Column(String, nullable=False)
|
||||
s3_bucket = Column(String, default="auteur-assets")
|
||||
thumbnail_key = Column(String, nullable=True)
|
||||
metadata_ = Column("metadata", JSONB, default={}) # 'metadata' is reserved in SQLAlchemy Base
|
||||
created_at = Column(DateTime, default=func.now())
|
||||
|
||||
project = relationship("Project", back_populates="ingredients")
|
||||
21
backend/app/models/project.py
Normal file
21
backend/app/models/project.py
Normal file
@@ -0,0 +1,21 @@
|
||||
|
||||
from sqlalchemy import Column, String, DateTime, func, Text
|
||||
from sqlalchemy.dialects.postgresql import UUID
|
||||
import uuid
|
||||
from sqlalchemy.orm import relationship
|
||||
|
||||
from app.db.session import Base
|
||||
|
||||
class Project(Base):
|
||||
__tablename__ = "projects"
|
||||
|
||||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||||
name = Column(String, nullable=False)
|
||||
resolution = Column(String, default="4K")
|
||||
aspect_ratio = Column(String, default="16:9")
|
||||
veo_version = Column(String, default="3.1")
|
||||
created_at = Column(DateTime, default=func.now())
|
||||
updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
|
||||
|
||||
ingredients = relationship("Ingredient", back_populates="project", cascade="all, delete-orphan")
|
||||
scenes = relationship("Scene", back_populates="project", cascade="all, delete-orphan")
|
||||
20
backend/app/models/scene.py
Normal file
20
backend/app/models/scene.py
Normal file
@@ -0,0 +1,20 @@
|
||||
|
||||
from sqlalchemy import Column, String, Integer, DateTime, func, ForeignKey, Text
|
||||
from sqlalchemy.dialects.postgresql import UUID
|
||||
import uuid
|
||||
from sqlalchemy.orm import relationship
|
||||
|
||||
from app.db.session import Base
|
||||
|
||||
class Scene(Base):
|
||||
__tablename__ = "scenes"
|
||||
|
||||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||||
project_id = Column(UUID(as_uuid=True), ForeignKey("projects.id", ondelete="CASCADE"), nullable=False)
|
||||
slugline = Column(String, nullable=False)
|
||||
raw_content = Column(Text, nullable=True)
|
||||
sequence_number = Column(Integer, nullable=False)
|
||||
created_at = Column(DateTime, default=func.now())
|
||||
|
||||
project = relationship("Project", back_populates="scenes")
|
||||
shots = relationship("Shot", back_populates="scene", cascade="all, delete-orphan")
|
||||
30
backend/app/models/shot.py
Normal file
30
backend/app/models/shot.py
Normal file
@@ -0,0 +1,30 @@
|
||||
|
||||
from sqlalchemy import Column, String, Float, Integer, DateTime, func, ForeignKey, Text
|
||||
from sqlalchemy.dialects.postgresql import UUID, JSONB
|
||||
import uuid
|
||||
from sqlalchemy.orm import relationship
|
||||
|
||||
from app.db.session import Base
|
||||
|
||||
class Shot(Base):
|
||||
__tablename__ = "shots"
|
||||
|
||||
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||||
scene_id = Column(UUID(as_uuid=True), ForeignKey("scenes.id", ondelete="CASCADE"), nullable=False)
|
||||
description = Column(Text, nullable=False)
|
||||
duration = Column(Float, nullable=True)
|
||||
sequence_number = Column(Integer, nullable=True)
|
||||
|
||||
# Slot system: list of ingredient UUIDs
|
||||
assigned_ingredients = Column(JSONB, default=[])
|
||||
|
||||
# Context cache for debugging
|
||||
llm_context_cache = Column(Text, nullable=True)
|
||||
|
||||
# Final Veo payload
|
||||
veo_json_payload = Column(JSONB, nullable=True)
|
||||
|
||||
status = Column(String, default="draft") # draft, generating, ready
|
||||
updated_at = Column(DateTime, default=func.now(), onupdate=func.now())
|
||||
|
||||
scene = relationship("Scene", back_populates="shots")
|
||||
Reference in New Issue
Block a user