68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
import sys
|
||
import os
|
||
from app import create_app, db
|
||
from flask_migrate import upgrade, migrate, init, stamp
|
||
|
||
def init_migrations():
|
||
"""初始化迁移环境 (如果 migrations 文件夹不存在)"""
|
||
if not os.path.exists('migrations'):
|
||
print("📂 初始化 migrations 文件夹...")
|
||
with app.app_context():
|
||
init()
|
||
else:
|
||
print("✅ migrations 文件夹已存在，跳过初始化。")
|
||
|
||
def make_migrations(message="Auto update"):
|
||
"""生成迁移脚本 (检测 models.py 的变动)"""
|
||
print(f"📝 正在生成迁移脚本 (备注: {message})...")
|
||
with app.app_context():
|
||
try:
|
||
migrate(message=message)
|
||
print("✅ 迁移脚本生成成功！")
|
||
except Exception as e:
|
||
print(f"⚠️ 生成失败 (可能是没有变动): {e}")
|
||
|
||
def apply_migrations():
|
||
"""应用迁移到数据库 (执行 SQL)"""
|
||
print("🚀 正在将变动应用到数据库...")
|
||
with app.app_context():
|
||
upgrade()
|
||
print("✅ 数据库结构已同步至最新版！")
|
||
|
||
def force_sync():
|
||
"""强制同步 (不仅是 Upgrade，还包括 create_all)"""
|
||
print("🔧 正在进行全量同步检查...")
|
||
with app.app_context():
|
||
db.create_all()
|
||
# 标记当前状态为最新
|
||
stamp()
|
||
print("✅ 数据库表结构已确保存在。")
|
||
|
||
if __name__ == '__main__':
|
||
app = create_app()
|
||
|
||
if len(sys.argv) < 2:
|
||
print("\n🔧 数据库管理脚本使用说明：")
|
||
print(" python db_manager.py init -> 初始化迁移环境")
|
||
print(" python db_manager.py make -> 生成迁移文件 (检测 models.py 变动)")
|
||
print(" python db_manager.py up -> 执行更新 (修改数据库表结构)")
|
||
print(" python db_manager.py sync -> 一键同步模式 (生成 + 执行)")
|
||
sys.exit(0)
|
||
|
||
cmd = sys.argv[1]
|
||
|
||
if cmd == 'init':
|
||
init_migrations()
|
||
elif cmd == 'make':
|
||
msg = sys.argv[2] if len(sys.argv) > 2 else "schema_update"
|
||
make_migrations(msg)
|
||
elif cmd == 'up':
|
||
apply_migrations()
|
||
elif cmd == 'sync':
|
||
# 一键傻瓜模式
|
||
init_migrations()
|
||
make_migrations("auto_sync")
|
||
apply_migrations()
|
||
else:
|
||
print("❌ 未知命令")
|