2026-04-29-21-51-19 - 全栈系统改造：FastAPI后端+SAM2+PostgreSQL+Redis+MinIO+前端Zustand重构

2026-04-29 22:17:25 +08:00
parent c8f8686097
commit fd4b5e5b3d
39 changed files with 3816 additions and 211 deletions
--- a/backend/models.py
+++ b/backend/models.py
@@ -0,0 +1,118 @@
+"""SQLAlchemy ORM models."""
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    Text,
+    DateTime,
+    ForeignKey,
+    JSON,
+    Float,
+)
+from sqlalchemy.orm import relationship
+from sqlalchemy.sql import func
+
+from database import Base
+
+
+class Project(Base):
+    """Project model representing a segmentation project."""
+
+    __tablename__ = "projects"
+
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String(255), nullable=False)
+    description = Column(Text, nullable=True)
+    video_path = Column(String(512), nullable=True)
+    status = Column(String(50), default="pending", nullable=False)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+    updated_at = Column(
+        DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
+    )
+
+    frames = relationship("Frame", back_populates="project", cascade="all, delete-orphan")
+    annotations = relationship(
+        "Annotation", back_populates="project", cascade="all, delete-orphan"
+    )
+
+
+class Frame(Base):
+    """Frame model representing an extracted video frame."""
+
+    __tablename__ = "frames"
+
+    id = Column(Integer, primary_key=True, index=True)
+    project_id = Column(Integer, ForeignKey("projects.id", ondelete="CASCADE"), nullable=False)
+    frame_index = Column(Integer, nullable=False)
+    image_url = Column(String(512), nullable=False)
+    width = Column(Integer, nullable=True)
+    height = Column(Integer, nullable=True)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+
+    project = relationship("Project", back_populates="frames")
+    annotations = relationship(
+        "Annotation", back_populates="frame", cascade="all, delete-orphan"
+    )
+
+
+class Template(Base):
+    """Template (Ontology) model for segmentation classes."""
+
+    __tablename__ = "templates"
+
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String(255), nullable=False)
+    color = Column(String(50), nullable=False)
+    z_index = Column(Integer, default=0, nullable=False)
+    mapping_rules = Column(JSON, nullable=True)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+
+    annotations = relationship(
+        "Annotation", back_populates="template", cascade="all, delete-orphan"
+    )
+
+
+class Annotation(Base):
+    """Annotation model for segmentation masks and prompts."""
+
+    __tablename__ = "annotations"
+
+    id = Column(Integer, primary_key=True, index=True)
+    project_id = Column(
+        Integer, ForeignKey("projects.id", ondelete="CASCADE"), nullable=False
+    )
+    frame_id = Column(
+        Integer, ForeignKey("frames.id", ondelete="CASCADE"), nullable=True
+    )
+    template_id = Column(
+        Integer, ForeignKey("templates.id", ondelete="SET NULL"), nullable=True
+    )
+    mask_data = Column(JSON, nullable=True)
+    points = Column(JSON, nullable=True)
+    bbox = Column(JSON, nullable=True)
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+    updated_at = Column(
+        DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
+    )
+
+    project = relationship("Project", back_populates="annotations")
+    frame = relationship("Frame", back_populates="annotations")
+    template = relationship("Template", back_populates="annotations")
+    masks = relationship("Mask", back_populates="annotation", cascade="all, delete-orphan")
+
+
+class Mask(Base):
+    """Mask model for exported/derived mask files."""
+
+    __tablename__ = "masks"
+
+    id = Column(Integer, primary_key=True, index=True)
+    annotation_id = Column(
+        Integer, ForeignKey("annotations.id", ondelete="CASCADE"), nullable=False
+    )
+    mask_url = Column(String(512), nullable=False)
+    format = Column(String(50), default="png", nullable=False)  # png / rle / json
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+
+    annotation = relationship("Annotation", back_populates="masks")