init: 初始化项目

backend/app/models/__init__.py

@@ -0,0 +1,17 @@
+from app.models.base import Base, TimestampMixin, UUIDMixin
+from app.models.ai_model import AIModel
+from app.models.template import Template
+from app.models.generation_point import GenerationPoint
+from app.models.generation_task import GenerationTask
+from app.models.system_config import SystemConfig
+
+__all__ = [
+    "Base",
+    "TimestampMixin",
+    "UUIDMixin",
+    "AIModel",
+    "Template",
+    "GenerationPoint",
+    "GenerationTask",
+    "SystemConfig",
+]

backend/app/models/ai_model.py

@@ -0,0 +1,18 @@
+import uuid
+from sqlalchemy import Boolean, String, Text
+from sqlalchemy.dialects.postgresql import UUID, JSONB
+from sqlalchemy.orm import Mapped, mapped_column
+from app.core.database import Base
+from app.models.base import TimestampMixin, UUIDMixin
+
+
+class AIModel(Base, UUIDMixin, TimestampMixin):
+    __tablename__ = "models"
+
+    name: Mapped[str] = mapped_column(String(100), nullable=False)
+    provider: Mapped[str] = mapped_column(String(50), nullable=False)
+    endpoint: Mapped[str] = mapped_column(String(255), nullable=False)
+    api_key: Mapped[str] = mapped_column(Text, nullable=False)
+    extra_params: Mapped[dict] = mapped_column(JSONB, default=dict, nullable=False)
+    is_enabled: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
+    remark: Mapped[str | None] = mapped_column(Text, nullable=True)

backend/app/models/base.py

@@ -0,0 +1,17 @@
+import uuid
+from datetime import datetime
+from sqlalchemy import DateTime, func
+from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.orm import Mapped, mapped_column
+from app.core.database import Base
+
+
+class TimestampMixin:
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
+    )
+
+
+class UUIDMixin:
+    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)

backend/app/models/generation_point.py

@@ -0,0 +1,27 @@
+import uuid
+from sqlalchemy import Integer, String, Text, ForeignKey
+from sqlalchemy.dialects.postgresql import UUID, JSONB
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+from app.core.database import Base
+from app.models.base import TimestampMixin, UUIDMixin
+
+
+class GenerationPoint(Base, UUIDMixin, TimestampMixin):
+    __tablename__ = "generation_points"
+
+    template_id: Mapped[uuid.UUID] = mapped_column(
+        UUID(as_uuid=True), ForeignKey("templates.id", ondelete="CASCADE"), nullable=False, index=True
+    )
+    position: Mapped[dict] = mapped_column(JSONB, nullable=False)
+    prompt: Mapped[str] = mapped_column(Text, nullable=False)
+    model_id: Mapped[uuid.UUID | None] = mapped_column(
+        UUID(as_uuid=True), ForeignKey("models.id", ondelete="SET NULL"), nullable=True
+    )
+    ref_file_path: Mapped[str | None] = mapped_column(Text, nullable=True)
+    need_ref_file: Mapped[bool] = mapped_column(default=False, nullable=False)
+    remark: Mapped[str | None] = mapped_column(Text, nullable=True)
+    order: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
+    selected_text: Mapped[str | None] = mapped_column(Text, nullable=True)
+
+    template = relationship("Template")
+    model = relationship("AIModel")

backend/app/models/generation_task.py

@@ -0,0 +1,23 @@
+import uuid
+from datetime import datetime
+from sqlalchemy import String, Text, DateTime, ForeignKey
+from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+from app.core.database import Base
+from app.models.base import UUIDMixin
+
+
+class GenerationTask(Base, UUIDMixin):
+    __tablename__ = "generation_tasks"
+
+    template_id: Mapped[uuid.UUID] = mapped_column(
+        UUID(as_uuid=True), ForeignKey("templates.id", ondelete="CASCADE"), nullable=False, index=True
+    )
+    status: Mapped[str] = mapped_column(String(20), nullable=False, default="pending")
+    result_file_path: Mapped[str | None] = mapped_column(String(500), nullable=True)
+    error_msg: Mapped[str | None] = mapped_column(Text, nullable=True)
+    celery_task_id: Mapped[str | None] = mapped_column(String(100), nullable=True)
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True))
+    finished_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
+
+    template = relationship("Template")

backend/app/models/system_config.py

@@ -0,0 +1,11 @@
+from sqlalchemy import String, Text
+from sqlalchemy.orm import Mapped, mapped_column
+from app.core.database import Base
+
+
+class SystemConfig(Base):
+    __tablename__ = "system_config"
+
+    key: Mapped[str] = mapped_column(String(50), primary_key=True)
+    value: Mapped[str] = mapped_column(Text, nullable=True)
+    description: Mapped[str | None] = mapped_column(String(200), nullable=True)

backend/app/models/template.py

@@ -0,0 +1,14 @@
+import uuid
+from sqlalchemy import String, Text
+from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.orm import Mapped, mapped_column
+from app.core.database import Base
+from app.models.base import TimestampMixin, UUIDMixin
+
+
+class Template(Base, UUIDMixin, TimestampMixin):
+    __tablename__ = "templates"
+
+    name: Mapped[str] = mapped_column(String(200), nullable=False)
+    file_path: Mapped[str] = mapped_column(String(500), nullable=False)
+    html_content: Mapped[str | None] = mapped_column(Text, nullable=True)