feature: 新增伏笔管理系统，支持可视化追踪、AI智能关联回收及章节生成时的伏笔提醒

2026-01-19 17:24:37 +08:00
parent 927072d16f
commit 5f25deb289
19 changed files with 4068 additions and 91 deletions
@@ -16,6 +16,7 @@ from app.models.user import User, UserPassword
 from app.models.regeneration_task import RegenerationTask
 from app.models.career import Career, CharacterCareer
 from app.models.prompt_template import PromptTemplate
+from app.models.foreshadow import Foreshadow

 __all__ = [
    "Project",
@@ -40,5 +41,6 @@ __all__ = [
    "RegenerationTask",
    "Career",
    "CharacterCareer",
-    "PromptTemplate"
+    "PromptTemplate",
+    "Foreshadow"
 ]
@@ -0,0 +1,178 @@
+"""伏笔管理数据模型 - 独立管理小说伏笔的埋入和回收"""
+from sqlalchemy import Column, String, Text, Integer, DateTime, ForeignKey, Float, JSON, Boolean
+from sqlalchemy.sql import func
+from sqlalchemy.orm import relationship
+from app.database import Base
+import uuid
+
+
+class Foreshadow(Base):
+    """
+    伏笔管理表 - 独立管理小说伏笔
+    
+    支持以下功能：
+    1. 从章节分析结果自动同步伏笔
+    2. 用户手动添加自定义伏笔
+    3. 关联埋入章节和计划回收章节
+    4. 长线伏笔管理
+    5. 章节生成时的伏笔提醒
+    """
+    __tablename__ = "foreshadows"
+    
+    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
+    project_id = Column(String(36), ForeignKey("projects.id", ondelete="CASCADE"), nullable=False, index=True)
+    
+    # === 伏笔内容 ===
+    title = Column(String(200), nullable=False, comment="伏笔标题")
+    content = Column(Text, nullable=False, comment="伏笔详细内容/描述")
+    hint_text = Column(Text, comment="埋伏笔时的暗示文本(原文摘录或概述)")
+    resolution_text = Column(Text, comment="回收伏笔时的揭示文本(原文摘录或概述)")
+    
+    # === 来源信息 ===
+    source_type = Column(String(20), default='manual', comment="来源类型: analysis=分析提取, manual=手动添加")
+    source_memory_id = Column(String(100), comment="来源记忆ID(如从分析结果同步)")
+    source_analysis_id = Column(String(36), comment="来源分析任务ID")
+    
+    # === 章节关联 ===
+    # 埋入章节
+    plant_chapter_id = Column(String(36), ForeignKey("chapters.id", ondelete="SET NULL"), comment="埋入章节ID")
+    plant_chapter_number = Column(Integer, comment="埋入章节号(冗余存储便于查询)")
+    
+    # 计划回收章节
+    target_resolve_chapter_id = Column(String(36), ForeignKey("chapters.id", ondelete="SET NULL"), comment="计划回收章节ID")
+    target_resolve_chapter_number = Column(Integer, comment="计划回收章节号")
+    
+    # 实际回收章节
+    actual_resolve_chapter_id = Column(String(36), ForeignKey("chapters.id", ondelete="SET NULL"), comment="实际回收章节ID")
+    actual_resolve_chapter_number = Column(Integer, comment="实际回收章节号")
+    
+    # === 状态管理 ===
+    status = Column(String(20), default='pending', index=True, comment="""
+    伏笔状态:
+    - pending: 待埋入(已规划但未写入章节)
+    - planted: 已埋入(已在章节中埋下)
+    - resolved: 已回收(已在章节中回收)
+    - partially_resolved: 部分回收(长线伏笔可能分多次回收)
+    - abandoned: 已废弃(决定不再使用此伏笔)
+    """)
+    
+    is_long_term = Column(Boolean, default=False, comment="是否长线伏笔(跨多章的重要伏笔)")
+    
+    # === 重要性和优先级 ===
+    importance = Column(Float, default=0.5, comment="重要性评分 0.0-1.0")
+    strength = Column(Integer, default=5, comment="伏笔强度 1-10(影响读者多强烈)")
+    subtlety = Column(Integer, default=5, comment="隐藏度 1-10(越高越隐蔽)")
+    urgency = Column(Integer, default=0, comment="紧急度: 0=不紧急, 1=需关注, 2=急需回收")
+    
+    # === 关联信息 ===
+    related_characters = Column(JSON, comment="关联角色名列表: ['角色1', '角色2']")
+    related_foreshadow_ids = Column(JSON, comment="关联的其他伏笔ID列表(伏笔链)")
+    tags = Column(JSON, comment="标签列表: ['身世', '悬念', '反转']")
+    category = Column(String(50), comment="分类: identity(身世), mystery(悬念), item(物品), relationship(关系), event(事件)")
+    
+    # === 备注和说明 ===
+    notes = Column(Text, comment="创作备注(仅作者可见)")
+    resolution_notes = Column(Text, comment="回收方式说明")
+    
+    # === AI辅助设置 ===
+    auto_remind = Column(Boolean, default=True, comment="是否在章节生成时自动提醒")
+    remind_before_chapters = Column(Integer, default=5, comment="提前几章开始提醒回收")
+    include_in_context = Column(Boolean, default=True, comment="是否包含在生成上下文中")
+    
+    # === 时间戳 ===
+    created_at = Column(DateTime, server_default=func.now(), comment="创建时间")
+    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now(), comment="更新时间")
+    planted_at = Column(DateTime, comment="埋入时间")
+    resolved_at = Column(DateTime, comment="回收时间")
+    
+    def __repr__(self):
+        return f"<Foreshadow(id={self.id[:8]}, title={self.title}, status={self.status})>"
+    
+    def to_dict(self):
+        """转换为字典格式"""
+        return {
+            "id": self.id,
+            "project_id": self.project_id,
+            "title": self.title,
+            "content": self.content,
+            "hint_text": self.hint_text,
+            "resolution_text": self.resolution_text,
+            "source_type": self.source_type,
+            "source_memory_id": self.source_memory_id,
+            "plant_chapter_id": self.plant_chapter_id,
+            "plant_chapter_number": self.plant_chapter_number,
+            "target_resolve_chapter_id": self.target_resolve_chapter_id,
+            "target_resolve_chapter_number": self.target_resolve_chapter_number,
+            "actual_resolve_chapter_id": self.actual_resolve_chapter_id,
+            "actual_resolve_chapter_number": self.actual_resolve_chapter_number,
+            "status": self.status,
+            "is_long_term": self.is_long_term,
+            "importance": self.importance,
+            "strength": self.strength,
+            "subtlety": self.subtlety,
+            "urgency": self.urgency,
+            "related_characters": self.related_characters or [],
+            "related_foreshadow_ids": self.related_foreshadow_ids or [],
+            "tags": self.tags or [],
+            "category": self.category,
+            "notes": self.notes,
+            "resolution_notes": self.resolution_notes,
+            "auto_remind": self.auto_remind,
+            "remind_before_chapters": self.remind_before_chapters,
+            "include_in_context": self.include_in_context,
+            "created_at": self.created_at.isoformat() if self.created_at else None,
+            "updated_at": self.updated_at.isoformat() if self.updated_at else None,
+            "planted_at": self.planted_at.isoformat() if self.planted_at else None,
+            "resolved_at": self.resolved_at.isoformat() if self.resolved_at else None,
+        }
+    
+    def to_context_string(self) -> str:
+        """
+        转换为上下文字符串(用于章节生成提示)
+        """
+        parts = []
+        
+        # 基本信息
+        parts.append(f"伏笔「{self.title}」")
+        
+        # 埋入信息
+        if self.plant_chapter_number:
+            parts.append(f"(第{self.plant_chapter_number}章埋下)")
+        
+        # 内容摘要
+        content_preview = self.content[:100] if len(self.content) > 100 else self.content
+        parts.append(f": {content_preview}")
+        
+        # 计划回收
+        if self.target_resolve_chapter_number:
+            parts.append(f" [计划第{self.target_resolve_chapter_number}章回收]")
+        
+        # 关联角色
+        if self.related_characters:
+            parts.append(f" 涉及: {', '.join(self.related_characters[:3])}")
+        
+        return "".join(parts)
+    
+    def get_urgency_level(self, current_chapter: int) -> int:
+        """
+        计算当前紧急度
+        
+        Args:
+            current_chapter: 当前章节号
+        
+        Returns:
+            0=不紧急, 1=需关注, 2=急需回收, 3=已超期
+        """
+        if self.status != 'planted' or not self.target_resolve_chapter_number:
+            return 0
+        
+        chapters_remaining = self.target_resolve_chapter_number - current_chapter
+        
+        if chapters_remaining < 0:
+            return 3  # 已超期
+        elif chapters_remaining <= 2:
+            return 2  # 急需回收
+        elif chapters_remaining <= self.remind_before_chapters:
+            return 1  # 需关注
+        else:
+            return 0  # 不紧急