69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
|
"""迁移 029：扩容 PPT 模板资产标识并增加版本指纹。"""
|
|||
|
|
import logging
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
from sqlalchemy import inspect, text
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def migrate():
|
|||
|
|
from insurance.db.compat import db
|
|||
|
|
|
|||
|
|
columns = {
|
|||
|
|
item["name"]: item
|
|||
|
|
for item in inspect(db.engine).get_columns("insurance_ppt_templates")
|
|||
|
|
}
|
|||
|
|
dialect = db.engine.dialect.name
|
|||
|
|
|
|||
|
|
if dialect == "postgresql":
|
|||
|
|
db.session.execute(text(
|
|||
|
|
"ALTER TABLE insurance_ppt_templates "
|
|||
|
|
"ALTER COLUMN source_template_asset_id TYPE VARCHAR(255)"
|
|||
|
|
))
|
|||
|
|
elif dialect in {"mysql", "mariadb"}:
|
|||
|
|
db.session.execute(text(
|
|||
|
|
"ALTER TABLE insurance_ppt_templates "
|
|||
|
|
"MODIFY source_template_asset_id VARCHAR(255)"
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
if "asset_sha256" not in columns:
|
|||
|
|
db.session.execute(text(
|
|||
|
|
"ALTER TABLE insurance_ppt_templates ADD COLUMN asset_sha256 VARCHAR(64)"
|
|||
|
|
))
|
|||
|
|
if "asset_version" not in columns:
|
|||
|
|
db.session.execute(text(
|
|||
|
|
"ALTER TABLE insurance_ppt_templates "
|
|||
|
|
"ADD COLUMN asset_version INTEGER NOT NULL DEFAULT 1"
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
from insurance.ppt.template_asset_service import (
|
|||
|
|
resolve_template_asset,
|
|||
|
|
template_asset_sha256,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
rows = db.session.execute(text(
|
|||
|
|
"SELECT id, source_template_asset_id FROM insurance_ppt_templates "
|
|||
|
|
"WHERE source_template_asset_id IS NOT NULL"
|
|||
|
|
)).mappings().all()
|
|||
|
|
for row in rows:
|
|||
|
|
try:
|
|||
|
|
path = resolve_template_asset(row["source_template_asset_id"])
|
|||
|
|
digest = template_asset_sha256(path) if path and os.path.isfile(path) else None
|
|||
|
|
except (OSError, ValueError):
|
|||
|
|
digest = None
|
|||
|
|
if digest:
|
|||
|
|
db.session.execute(text(
|
|||
|
|
"UPDATE insurance_ppt_templates "
|
|||
|
|
"SET asset_sha256 = :digest, asset_version = COALESCE(asset_version, 1) "
|
|||
|
|
"WHERE id = :id"
|
|||
|
|
), {"digest": digest, "id": row["id"]})
|
|||
|
|
|
|||
|
|
db.session.execute(text(
|
|||
|
|
"UPDATE insurance_ppt_templates SET clone_renderer = 'clone-edit-v2' "
|
|||
|
|
"WHERE source_template_asset_id IS NOT NULL"
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
db.session.commit()
|
|||
|
|
logger.info("[migrate_029] PPT 模板资产字段与指纹迁移完成")
|