124 lines
4.4 KiB
Python
124 lines
4.4 KiB
Python
"""从 `alembic/baseline_generated.sql` 生成基线初始迁移（方案 A）。
|
|
|
|
背景：39 张基线表此前没有任何 Alembic 迁移，空库无法重建。本脚本把基线 DDL 转成一条
|
|
初始迁移（`down_revision=None`），使 `alembic upgrade head` 能在空库上建出全部业务表。
|
|
|
|
设计取舍：迁移体内使用 `op.execute` 原样执行基线 SQL，而不是手工翻译成 `op.create_table`。
|
|
39 张表包含生成列、字符集与排序规则、联合唯一键等方言细节，逐表翻译容易引入新的结构漂移；
|
|
原样执行可保证"迁移产生的结构与 baseline_generated.sql 一致"，而该文件由
|
|
`tools/generate_baseline_sql.py` 从基线文档生成，并由 `tools/audit_constraints.py` 审计。
|
|
|
|
默认只打印摘要，加 `--write` 才落盘，避免在 Alembic 链上意外产生双 head。
|
|
|
|
用法：
|
|
python tools/generate_baseline_migration.py # 预览
|
|
python tools/generate_baseline_migration.py --write # 写入迁移文件
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
SQL_FILE = ROOT / "alembic" / "baseline_generated.sql"
|
|
TARGET = ROOT / "alembic" / "versions" / "20260909_baseline_schema.py"
|
|
REVISION = "20260909_baseline_schema"
|
|
|
|
TABLE_RE = re.compile(r"CREATE TABLE\s+`?([A-Za-z0-9_]+)`?")
|
|
|
|
HEADER = '''"""基线结构初始迁移（方案 A）：使空库可重建全部基线表。
|
|
|
|
本文件由 `tools/generate_baseline_migration.py` 从 `alembic/baseline_generated.sql` 生成，
|
|
请勿手工编辑；结构变更请改 `docs/00-新数据库基线设计.md` 后重新生成基线 SQL 与迁移。
|
|
|
|
设计说明：基线表此前不在 Alembic 链上，导致空库无法重建（迁移链只创建平台增量表）。
|
|
本迁移作为链首（`down_revision=None`）：既有库的 `alembic_version` 已位于本迁移的后代，
|
|
Alembic 不会重复执行；空库则从本迁移开始依次建表。
|
|
|
|
约束：只创建基线表，不修改任何既有表名或既有字段定义。
|
|
"""
|
|
from alembic import op
|
|
|
|
revision = "{revision}"
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
STATEMENTS: list[str] = [
|
|
'''
|
|
|
|
TAIL = '''
|
|
|
|
def upgrade() -> None:
|
|
for statement in STATEMENTS:
|
|
op.execute(statement)
|
|
|
|
|
|
def downgrade() -> None:
|
|
# 仅用于整体回退到空库；真实环境回退请使用前向兼容迁移。
|
|
for table in reversed(TABLES):
|
|
op.execute(f"DROP TABLE IF EXISTS `{table}`")
|
|
'''
|
|
|
|
|
|
def split_statements(sql_text: str) -> list[str]:
|
|
statements: list[str] = []
|
|
current: list[str] = []
|
|
for line in sql_text.splitlines():
|
|
stripped = line.strip()
|
|
if not stripped or stripped.startswith("--"):
|
|
continue
|
|
if stripped.startswith("SET FOREIGN_KEY_CHECKS"):
|
|
continue
|
|
current.append(line)
|
|
if stripped.endswith(";"):
|
|
statements.append("\n".join(current).rstrip().rstrip(";"))
|
|
current = []
|
|
if current:
|
|
statements.append("\n".join(current).rstrip())
|
|
return statements
|
|
|
|
|
|
def table_names(statements: list[str]) -> list[str]:
|
|
names: list[str] = []
|
|
for statement in statements:
|
|
match = TABLE_RE.match(statement.strip())
|
|
if match:
|
|
names.append(match.group(1))
|
|
return names
|
|
|
|
|
|
def render(statements: list[str], tables: list[str]) -> str:
|
|
body = "".join(
|
|
' """' + statement.replace('"""', '\\"\\"\\"') + '""",\n' for statement in statements
|
|
)
|
|
tables_block = "TABLES: list[str] = [\n" + "".join(
|
|
f' "{name}",\n' for name in tables
|
|
) + "]\n"
|
|
return HEADER.format(revision=REVISION) + body + "]\n\n" + tables_block + TAIL
|
|
|
|
|
|
def main() -> int:
|
|
if not SQL_FILE.exists():
|
|
print(f"baseline sql missing: {SQL_FILE}")
|
|
return 1
|
|
statements = split_statements(SQL_FILE.read_text(encoding="utf-8"))
|
|
tables = table_names(statements)
|
|
if len(statements) != len(tables):
|
|
print(f"unexpected: statements={len(statements)} tables={len(tables)}")
|
|
return 1
|
|
if "--write" not in sys.argv:
|
|
print(f"preview: statements={len(statements)} tables={len(tables)}")
|
|
print(f"target : {TARGET}")
|
|
print("run with --write to create the migration file")
|
|
return 0
|
|
TARGET.write_text(render(statements, tables), encoding="utf-8")
|
|
print(f"written: {TARGET} (statements={len(statements)}, tables={len(tables)})")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|