31 lines
1.8 KiB
Python
31 lines
1.8 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="原始锚点引用")
|
|
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())
|