61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""幂等迁移：为 ``escrows`` 表补充微信服务商分账扩展列（非运行态）。
|
|||
|
|
|
|||
|
|
用法：uv run python scripts/db/migrate_escrow_columns.py
|
|||
|
|
|
|||
|
|
- 通过 SQLAlchemy Inspector 检查缺失列，对缺失列执行 ``ALTER TABLE ... ADD COLUMN``；
|
|||
|
|
- 幂等：已存在列自动跳过，可重复执行；兼容 SQLite（默认）与 MySQL（生产）。
|
|||
|
|
- 新增表 ``payment_bindings`` / ``settlement_logs`` 由应用启动时 ``create_all`` 幂等创建，
|
|||
|
|
无需在此处理（本脚本只负责对既有表加列）。
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
ROOT = Path(__file__).resolve().parent.parent.parent
|
|||
|
|
sys.path.insert(0, str(ROOT))
|
|||
|
|
|
|||
|
|
from sqlalchemy import create_engine, inspect # noqa: E402
|
|||
|
|
|
|||
|
|
from app.infrastructure.db import sync_database_url # noqa: E402
|
|||
|
|
|
|||
|
|
# escrows 需补充的扩展列（列名, 方言通用 DDL）
|
|||
|
|
ESCROW_COLUMNS = [
|
|||
|
|
("channel", "VARCHAR(32) NOT NULL DEFAULT 'wx_split'"),
|
|||
|
|
("transaction_id", "VARCHAR(64) NOT NULL DEFAULT ''"),
|
|||
|
|
("share_order_no", "VARCHAR(64) NOT NULL DEFAULT ''"),
|
|||
|
|
("share_status", "VARCHAR(32) NOT NULL DEFAULT 'pending'"),
|
|||
|
|
("receiver_binding_id", "VARCHAR(64) NOT NULL DEFAULT ''"),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> None:
|
|||
|
|
url = sync_database_url()
|
|||
|
|
engine = create_engine(url)
|
|||
|
|
insp = inspect(engine)
|
|||
|
|
if "escrows" not in insp.get_table_names():
|
|||
|
|
print("escrows 表不存在（未建库），无需迁移")
|
|||
|
|
return
|
|||
|
|
existing = {c["name"] for c in insp.get_columns("escrows")}
|
|||
|
|
added, skipped = [], []
|
|||
|
|
for name, ddl in ESCROW_COLUMNS:
|
|||
|
|
if name in existing:
|
|||
|
|
skipped.append(name)
|
|||
|
|
continue
|
|||
|
|
with engine.begin() as conn:
|
|||
|
|
conn.exec_driver_sql(f"ALTER TABLE escrows ADD COLUMN {name} {ddl}")
|
|||
|
|
added.append(name)
|
|||
|
|
print(f"新增列: {added or '无'}")
|
|||
|
|
print(f"已存在跳过: {skipped or '无'}")
|
|||
|
|
engine.dispose()
|
|||
|
|
if added:
|
|||
|
|
print("迁移完成。")
|
|||
|
|
else:
|
|||
|
|
print("无新增列，escrows 已是最新。")
|
|||
|
|
print("注：新表 payment_bindings / settlement_logs 由应用启动时 create_all 幂等创建。")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|