MuMuAINovel/backend/app/models/project.py

"""项目数据模型"""
from sqlalchemy import Column, String, Text, DateTime, Integer, CheckConstraint
from sqlalchemy.sql import func
from app.database import Base
import uuid


class Project(Base):
    """项目表"""
    __tablename__ = "projects"
    
    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    user_id = Column(String(100), nullable=False, index=True, comment="用户ID")
    title = Column(String(200), nullable=False, comment="项目标题")
    description = Column(Text, comment="项目简介")
    theme = Column(Text, comment="主题")
    genre = Column(String(50), comment="小说类型")
    target_words = Column(Integer, default=0, comment="目标字数")
    current_words = Column(Integer, default=0, comment="当前字数")
    status = Column(String(20), default="planning", comment="创作状态")
    wizard_status = Column(String(20), default="incomplete", comment="向导完成状态: incomplete/completed")
    wizard_step = Column(Integer, default=0, comment="向导当前步骤: 0-4")
    outline_mode = Column(String(20), nullable=False, default="one-to-many", comment="大纲章节模式: one-to-one(传统模式) 或 one-to-many(细化模式)")
    
    # 世界构建字段
    world_time_period = Column(Text, comment="时间背景")
    world_location = Column(Text, comment="地理位置")
    world_atmosphere = Column(Text, comment="氛围基调")
    world_rules = Column(Text, comment="世界规则")
    
    # 项目配置
    chapter_count = Column(Integer, comment="章节数量")
    narrative_perspective = Column(String(50), comment="叙事视角：first_person/third_person/omniscient")
    character_count = Column(Integer, default=5, comment="角色数量")
    
    created_at = Column(DateTime, server_default=func.now(), comment="创建时间")
    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now(), comment="更新时间")
    
    __table_args__ = (
        CheckConstraint(
            "outline_mode IN ('one-to-one', 'one-to-many')",
            name='check_outline_mode'
        ),
    )
    
    def __repr__(self):
        return f"<Project(id={self.id}, title={self.title})>"