101 lines
3.9 KiB
Python
101 lines
3.9 KiB
Python
|
"""迁移 026：用户产品小册子资料库与海报产品来源快照。"""
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
from sqlalchemy import inspect, text
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _primary_key_sql(dialect: str) -> str:
|
|||
|
|
if dialect == "postgresql":
|
|||
|
|
return "BIGSERIAL PRIMARY KEY"
|
|||
|
|
if dialect in ("mysql", "mariadb"):
|
|||
|
|
return "BIGINT AUTO_INCREMENT PRIMARY KEY"
|
|||
|
|
return "INTEGER PRIMARY KEY AUTOINCREMENT"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _column_names(db, table_name: str) -> set[str]:
|
|||
|
|
return {item["name"] for item in inspect(db.engine).get_columns(table_name)}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _index_names(db, table_name: str) -> set[str]:
|
|||
|
|
return {item["name"] for item in inspect(db.engine).get_indexes(table_name)}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def migrate():
|
|||
|
|
"""幂等创建用户资料表并扩展海报关联字段。"""
|
|||
|
|
from insurance.db.compat import db
|
|||
|
|
|
|||
|
|
dialect = db.engine.dialect.name
|
|||
|
|
inspector = inspect(db.engine)
|
|||
|
|
if not inspector.has_table("insurance_user_product_materials"):
|
|||
|
|
db.session.execute(text(f"""
|
|||
|
|
CREATE TABLE insurance_user_product_materials (
|
|||
|
|
id {_primary_key_sql(dialect)},
|
|||
|
|
owner_user_id VARCHAR(64) NOT NULL,
|
|||
|
|
tenant_id VARCHAR(64) NOT NULL DEFAULT 'default',
|
|||
|
|
company_id VARCHAR(50),
|
|||
|
|
company_name VARCHAR(100),
|
|||
|
|
product_name VARCHAR(150),
|
|||
|
|
plan_type VARCHAR(20),
|
|||
|
|
original_name VARCHAR(200) NOT NULL,
|
|||
|
|
file_key VARCHAR(500) NOT NULL,
|
|||
|
|
file_size BIGINT NOT NULL DEFAULT 0,
|
|||
|
|
page_count INTEGER,
|
|||
|
|
sha256 VARCHAR(64) NOT NULL,
|
|||
|
|
parse_status VARCHAR(20) NOT NULL DEFAULT 'pending',
|
|||
|
|
parse_message VARCHAR(500) NOT NULL DEFAULT '',
|
|||
|
|
parse_error TEXT,
|
|||
|
|
parse_task_id VARCHAR(200),
|
|||
|
|
parse_started_at TIMESTAMP,
|
|||
|
|
parse_finished_at TIMESTAMP,
|
|||
|
|
parsed_rules TEXT,
|
|||
|
|
confirmed_rules TEXT,
|
|||
|
|
confirmed_by VARCHAR(64),
|
|||
|
|
confirmed_at TIMESTAMP,
|
|||
|
|
review_status VARCHAR(20) NOT NULL DEFAULT 'private',
|
|||
|
|
status SMALLINT NOT NULL DEFAULT 1,
|
|||
|
|
deleted_at TIMESTAMP,
|
|||
|
|
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
|||
|
|
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|||
|
|
)
|
|||
|
|
"""))
|
|||
|
|
db.session.commit()
|
|||
|
|
logger.info("[migrate_026] 已创建 insurance_user_product_materials")
|
|||
|
|
|
|||
|
|
desired_indexes = {
|
|||
|
|
"idx_user_material_owner_updated": "owner_user_id, updated_at",
|
|||
|
|
"idx_user_material_owner_sha256": "owner_user_id, sha256",
|
|||
|
|
"idx_user_material_parse_status": "parse_status",
|
|||
|
|
}
|
|||
|
|
existing_indexes = _index_names(db, "insurance_user_product_materials")
|
|||
|
|
for index_name, columns in desired_indexes.items():
|
|||
|
|
if index_name not in existing_indexes:
|
|||
|
|
db.session.execute(text(
|
|||
|
|
f"CREATE INDEX {index_name} "
|
|||
|
|
f"ON insurance_user_product_materials ({columns})"
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
extra_columns = {
|
|||
|
|
"product_source_type": "VARCHAR(20)",
|
|||
|
|
"product_source_id": "VARCHAR(64)",
|
|||
|
|
"product_snapshot_json": "TEXT",
|
|||
|
|
}
|
|||
|
|
for table_name in ("poster_case_uploads", "poster_records"):
|
|||
|
|
current_columns = _column_names(db, table_name)
|
|||
|
|
for column_name, column_type in extra_columns.items():
|
|||
|
|
if column_name not in current_columns:
|
|||
|
|
db.session.execute(text(
|
|||
|
|
f"ALTER TABLE {table_name} ADD COLUMN {column_name} {column_type}"
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
db.session.execute(text(
|
|||
|
|
f"UPDATE {table_name} "
|
|||
|
|
"SET product_source_type = 'library_product', product_source_id = product_id "
|
|||
|
|
"WHERE product_id IS NOT NULL AND product_id <> '' "
|
|||
|
|
"AND (product_source_type IS NULL OR product_source_type = '')"
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
db.session.commit()
|
|||
|
|
logger.info("[migrate_026] 用户产品资料库迁移完成")
|