52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
"""迁移启动脚本 - 用于执行数据库迁移
|
|||
|
|
|
|||
|
|
使用方法：
|
|||
|
|
python run_migration.py
|
|||
|
|
"""
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
# 添加项目根目录到Python路径
|
|||
|
|
project_root = os.path.abspath(os.path.dirname(__file__))
|
|||
|
|
sys.path.insert(0, project_root)
|
|||
|
|
|
|||
|
|
# 重要：先加载.env文件，再导入模块
|
|||
|
|
from lawrisk.utils.env_loader import load_env
|
|||
|
|
load_env('.env', override=False)
|
|||
|
|
|
|||
|
|
# 导入并执行迁移
|
|||
|
|
from lawrisk.utils.migrate_unit_permission import execute_migration, check_migration_status
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("许可管理单位权限优化 - 数据库迁移")
|
|||
|
|
print("=" * 60)
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
# 检查当前状态
|
|||
|
|
status = check_migration_status()
|
|||
|
|
print(f"迁移状态: {'已完成' if status['migration_complete'] else '未完成'}")
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
if not status['migration_complete']:
|
|||
|
|
response = input("是否执行数据库迁移？(y/N): ").strip().lower()
|
|||
|
|
if response == 'y':
|
|||
|
|
print("\n开始执行迁移...")
|
|||
|
|
results = execute_migration()
|
|||
|
|
|
|||
|
|
if results['success']:
|
|||
|
|
print("\n✅ 迁移成功完成！")
|
|||
|
|
print(" - 架构迁移: ✅")
|
|||
|
|
print(" - 数据迁移: ✅")
|
|||
|
|
else:
|
|||
|
|
print("\n❌ 迁移失败！")
|
|||
|
|
for error in results['errors']:
|
|||
|
|
print(f" 错误: {error}")
|
|||
|
|
else:
|
|||
|
|
print("\n跳过迁移")
|
|||
|
|
else:
|
|||
|
|
print("数据库已完全迁移，无需重复执行")
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 60)
|