doc-forge-remark/backend/models/template_block.py

34 lines
2.1 KiB
Python

from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, String, Text, func
from database import Base
class TemplateBlock(Base):
__tablename__ = "template_blocks"
id = Column(Integer, primary_key=True, autoincrement=True)
template_id = Column(Integer, ForeignKey("templates.id"), nullable=False)
source_paragraph_id = Column(Integer, ForeignKey("paragraphs.id"), nullable=True)
parent_block_id = Column(Integer, ForeignKey("template_blocks.id"), nullable=True)
sort_index = Column(Integer, default=0, comment="排序")
block_type = Column(String(30), default="text", comment="heading/text/table/ai_slot/variable")
anchor_ref = Column(String(500), default="", comment="原始锚点引用")
anchor_start_index = Column(Integer, default=0, comment="原始文档起始元素索引")
anchor_end_index = Column(Integer, default=0, comment="原始文档结束元素索引")
html_snippet = Column(Text, comment="渲染后的 HTML 片段")
title = Column(String(500), default="", comment="块标题")
content_json = Column(Text, default="{}", comment="块内容 JSON")
style_json = Column(Text, default="{}", comment="块样式 JSON")
edit_mode = Column(String(20), default="manual", comment="manual/ai")
placeholder_key = Column(String(120), default="", comment="AI 占位键")
variable_key = Column(String(120), default="", comment="变量键")
default_value = Column(Text, default="", comment="默认值")
model_id = Column(Integer, ForeignKey("ai_models.id"), nullable=True, comment="指定模型")
need_prompt = Column(Boolean, default=True, comment="是否需要提示词")
prompt_text = Column(Text, default="", comment="预设提示词")
need_file = Column(Boolean, default=False, comment="是否需要上传参考文件")
file_note = Column(Text, default="", comment="参考文件说明")
output_format = Column(String(20), default="text", comment="text/table/mixed/chart")
created_at = Column(DateTime, server_default=func.now())
updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now())