61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
数据库迁移脚本
|
|||
|
|
"""
|
|||
|
|
import asyncio
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
# 添加项目根目录到路径
|
|||
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|||
|
|
|
|||
|
|
async def main():
|
|||
|
|
from app.core.config import get_settings
|
|||
|
|
from aerich import Command
|
|||
|
|
|
|||
|
|
settings = get_settings()
|
|||
|
|
|
|||
|
|
# Aerich 配置
|
|||
|
|
TORTOISE_ORM = {
|
|||
|
|
"connections": {
|
|||
|
|
"default": settings.database_url
|
|||
|
|
},
|
|||
|
|
"apps": {
|
|||
|
|
"models": {
|
|||
|
|
"models": ["app.models", "aerich.models"],
|
|||
|
|
"default_connection": "default",
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
"use_tz": False,
|
|||
|
|
"timezone": "UTC",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
command = Command(tortoise_config=TORTOISE_ORM, app="models")
|
|||
|
|
|
|||
|
|
# 初始化 Aerich（首次运行）
|
|||
|
|
if not os.path.exists("./migrations"):
|
|||
|
|
print("🔧 初始化数据库迁移...")
|
|||
|
|
await command.init()
|
|||
|
|
print("✅ 迁移初始化完成")
|
|||
|
|
|
|||
|
|
# 生成迁移文件
|
|||
|
|
print("📝 生成迁移文件...")
|
|||
|
|
try:
|
|||
|
|
migration_name = sys.argv[1] if len(sys.argv) > 1 else "update"
|
|||
|
|
await command.migrate(name=migration_name)
|
|||
|
|
print(f"✅ 迁移文件生成成功: {migration_name}")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"⚠️ 生成迁移文件失败: {e}")
|
|||
|
|
# 尝试升级
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# 应用迁移
|
|||
|
|
print("🔄 应用数据库迁移...")
|
|||
|
|
try:
|
|||
|
|
await command.upgrade()
|
|||
|
|
print("✅ 迁移应用成功")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 迁移应用失败: {e}")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
asyncio.run(main())
|