35 lines
1.7 KiB
Python
35 lines
1.7 KiB
Python
|
"""海报模板模型（AI 生图方案）。"""
|
|||
|
|
from sqlalchemy import Column, String, Text, SmallInteger, BigInteger, TIMESTAMP, func
|
|||
|
|
from insurance.db.compat import db
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PosterTemplate(db.Model):
|
|||
|
|
"""海报模板表（AI 生图方案）。"""
|
|||
|
|
__tablename__ = "poster_templates"
|
|||
|
|
|
|||
|
|
id = Column(BigInteger, primary_key=True, autoincrement=True)
|
|||
|
|
name = Column(String(100), nullable=False, comment="模板名称")
|
|||
|
|
scenario_tag = Column(String(50), nullable=True, comment="场景标签")
|
|||
|
|
style_description = Column(Text, nullable=False, comment="AI 生图风格描述 prompt 片段")
|
|||
|
|
color_scheme = Column(Text, nullable=True, comment="配色方案 JSON")
|
|||
|
|
reference_image = Column(String(500), nullable=True, comment="参考图地址")
|
|||
|
|
preview_image = Column(String(500), nullable=True, comment="预览图地址")
|
|||
|
|
status = Column(SmallInteger, default=1, comment="1=启用, 0=停用")
|
|||
|
|
created_at = Column(TIMESTAMP, server_default=func.now())
|
|||
|
|
updated_at = Column(TIMESTAMP, server_default=func.now(), onupdate=func.now())
|
|||
|
|
|
|||
|
|
def to_dict(self):
|
|||
|
|
import json
|
|||
|
|
return {
|
|||
|
|
"id": self.id,
|
|||
|
|
"name": self.name,
|
|||
|
|
"scenarioTag": self.scenario_tag,
|
|||
|
|
"styleDescription": self.style_description,
|
|||
|
|
"colorScheme": json.loads(self.color_scheme) if self.color_scheme else None,
|
|||
|
|
"referenceImage": self.reference_image,
|
|||
|
|
"previewImage": self.preview_image,
|
|||
|
|
"status": self.status,
|
|||
|
|
"createdAt": self.created_at.isoformat() if self.created_at else None,
|
|||
|
|
"updatedAt": self.updated_at.isoformat() if self.updated_at else None,
|
|||
|
|
}
|