100 lines
4.6 KiB
Python
100 lines
4.6 KiB
Python
|
"""迁移 020: 初始化海报模板和文案模板种子数据。
|
|||
|
|
|
|||
|
|
仅在表为空时插入，不覆盖管理员已修改的数据。
|
|||
|
|
解决 POSTER-P0-02: 新环境没有海报基础数据。
|
|||
|
|
"""
|
|||
|
|
import json
|
|||
|
|
import logging
|
|||
|
|
from sqlalchemy import text
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
# 海报模板种子数据
|
|||
|
|
POSTER_TEMPLATES = [
|
|||
|
|
{
|
|||
|
|
"name": "专业商务风",
|
|||
|
|
"scenario_tag": "business",
|
|||
|
|
"style_description": "专业保险营销海报，深蓝色背景搭配金色点缀，突出公司品牌和产品数据，排版整洁大方，适合储蓄险和寿险产品",
|
|||
|
|
"color_scheme": json.dumps({"primary": "#1a3a5c", "accent": "#d4a843", "bg": "#f5f7fa"}, ensure_ascii=False),
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "温馨家庭风",
|
|||
|
|
"scenario_tag": "family",
|
|||
|
|
"style_description": "温馨家庭保障主题海报，暖色调渐变背景，包含家庭剪影元素，突出保障金额和家庭关怀，适合重疾险和家庭保障产品",
|
|||
|
|
"color_scheme": json.dumps({"primary": "#e8734a", "accent": "#f5a623", "bg": "#fef9f3"}, ensure_ascii=False),
|
|||
|
|
},
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
# 文案模板种子数据
|
|||
|
|
COPY_TEMPLATES = [
|
|||
|
|
{
|
|||
|
|
"name": "储蓄险标准文案",
|
|||
|
|
"scenario_tag": "savings",
|
|||
|
|
"content": "{{product_name}} — 稳健增值，安心传承\n\n年缴 {{annual_premium}} {{currency}}，缴费 {{coverage_period}}，让财富在时间中复利增长。\n\n{{feature_1_title}}：{{feature_1_summary}}\n{{feature_2_title}}：{{feature_2_summary}}\n\n立即咨询您的专属顾问，获取个性化方案 →",
|
|||
|
|
"variables": json.dumps([
|
|||
|
|
{"name": "product_name", "desc": "产品名称"},
|
|||
|
|
{"name": "annual_premium", "desc": "年缴保费"},
|
|||
|
|
{"name": "currency", "desc": "货币"},
|
|||
|
|
{"name": "coverage_period", "desc": "保障期限"},
|
|||
|
|
{"name": "feature_1_title", "desc": "产品亮点1标题"},
|
|||
|
|
{"name": "feature_1_summary", "desc": "产品亮点1摘要"},
|
|||
|
|
{"name": "feature_2_title", "desc": "产品亮点2标题"},
|
|||
|
|
{"name": "feature_2_summary", "desc": "产品亮点2摘要"},
|
|||
|
|
], ensure_ascii=False),
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"name": "重疾险标准文案",
|
|||
|
|
"scenario_tag": "ci",
|
|||
|
|
"content": "{{product_name}} — 守护家人，抵御风险\n\n{{age}} 岁投保，保障 {{coverage_period}}，年缴 {{annual_premium}} {{currency}}。\n\n{{feature_1_title}}：{{feature_1_summary}}\n{{feature_2_title}}：{{feature_2_summary}}\n\n为家人撑起保护伞，立即了解详情 →",
|
|||
|
|
"variables": json.dumps([
|
|||
|
|
{"name": "product_name", "desc": "产品名称"},
|
|||
|
|
{"name": "age", "desc": "投保年龄"},
|
|||
|
|
{"name": "coverage_period", "desc": "保障期限"},
|
|||
|
|
{"name": "annual_premium", "desc": "年缴保费"},
|
|||
|
|
{"name": "currency", "desc": "货币"},
|
|||
|
|
{"name": "feature_1_title", "desc": "产品亮点1标题"},
|
|||
|
|
{"name": "feature_1_summary", "desc": "产品亮点1摘要"},
|
|||
|
|
{"name": "feature_2_title", "desc": "产品亮点2标题"},
|
|||
|
|
{"name": "feature_2_summary", "desc": "产品亮点2摘要"},
|
|||
|
|
], ensure_ascii=False),
|
|||
|
|
},
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _table_has_rows(db, table_name: str) -> bool:
|
|||
|
|
"""检查表是否有数据。"""
|
|||
|
|
try:
|
|||
|
|
result = db.session.execute(text(f"SELECT COUNT(*) FROM {table_name}"))
|
|||
|
|
return result.scalar() > 0
|
|||
|
|
except Exception:
|
|||
|
|
return True # 表不存在时返回 True，避免插入
|
|||
|
|
|
|||
|
|
|
|||
|
|
def migrate():
|
|||
|
|
"""执行迁移。"""
|
|||
|
|
from insurance.db.compat import db
|
|||
|
|
|
|||
|
|
# 海报模板种子
|
|||
|
|
if not _table_has_rows(db, "poster_templates"):
|
|||
|
|
for tpl in POSTER_TEMPLATES:
|
|||
|
|
db.session.execute(text(
|
|||
|
|
"INSERT INTO poster_templates (name, scenario_tag, style_description, color_scheme, status) "
|
|||
|
|
"VALUES (:name, :scenario_tag, :style_description, :color_scheme, 1)"
|
|||
|
|
), tpl)
|
|||
|
|
logger.info(f"已插入 {len(POSTER_TEMPLATES)} 个海报模板种子")
|
|||
|
|
else:
|
|||
|
|
logger.info("poster_templates 表已有数据，跳过种子插入")
|
|||
|
|
|
|||
|
|
# 文案模板种子
|
|||
|
|
if not _table_has_rows(db, "poster_copy_templates"):
|
|||
|
|
for tpl in COPY_TEMPLATES:
|
|||
|
|
db.session.execute(text(
|
|||
|
|
"INSERT INTO poster_copy_templates (name, scenario_tag, content, variables, status) "
|
|||
|
|
"VALUES (:name, :scenario_tag, :content, :variables, 1)"
|
|||
|
|
), tpl)
|
|||
|
|
logger.info(f"已插入 {len(COPY_TEMPLATES)} 个文案模板种子")
|
|||
|
|
else:
|
|||
|
|
logger.info("poster_copy_templates 表已有数据，跳过种子插入")
|
|||
|
|
|
|||
|
|
db.session.commit()
|