41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
"""迁移 031：增加海报模板布局与格式兼容字段。"""
|
||
import logging
|
||
|
||
from sqlalchemy import inspect, text
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
def migrate():
|
||
from insurance.db.compat import db
|
||
|
||
columns = {
|
||
item["name"]
|
||
for item in inspect(db.engine).get_columns("poster_templates")
|
||
}
|
||
additions = {
|
||
"layout_key": "VARCHAR(50)",
|
||
"supported_modes": "TEXT",
|
||
"supported_formats": "TEXT",
|
||
}
|
||
for name, column_type in additions.items():
|
||
if name not in columns:
|
||
db.session.execute(text(
|
||
f"ALTER TABLE poster_templates ADD COLUMN {name} {column_type}"
|
||
))
|
||
|
||
db.session.execute(text(
|
||
"UPDATE poster_templates SET layout_key = :layout "
|
||
"WHERE layout_key IS NULL OR layout_key = ''"
|
||
), {"layout": "legacy"})
|
||
db.session.execute(text(
|
||
"UPDATE poster_templates SET supported_modes = :modes "
|
||
"WHERE supported_modes IS NULL OR supported_modes = ''"
|
||
), {"modes": '["single","long"]'})
|
||
db.session.execute(text(
|
||
"UPDATE poster_templates SET supported_formats = :formats "
|
||
"WHERE supported_formats IS NULL OR supported_formats = ''"
|
||
), {"formats": '["single_2_3","single_9_16","long_1242_auto"]'})
|
||
db.session.commit()
|
||
logger.info("[migrate_031] 海报模板布局与格式兼容字段迁移完成")
|