Lan
81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
import glob
|
|
import importlib
|
|
import os
|
|
|
|
from tortoise import Tortoise
|
|
|
|
from core.settings import data_root
|
|
|
|
|
|
async def init_db():
|
|
try:
|
|
# 使用正确的Tortoise初始化配置格式
|
|
db_config = {
|
|
"db_url": f"sqlite://{data_root}/filecodebox.db",
|
|
"modules": {"models": ["apps.base.models"]},
|
|
"use_tz": False,
|
|
"timezone": "Asia/Shanghai"
|
|
}
|
|
|
|
await Tortoise.init(**db_config)
|
|
|
|
# 创建migrations表
|
|
await Tortoise.get_connection("default").execute_script("""
|
|
CREATE TABLE IF NOT EXISTS migrates (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
migration_file VARCHAR(255) NOT NULL UNIQUE,
|
|
executed_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
""")
|
|
|
|
# 执行迁移
|
|
await execute_migrations()
|
|
|
|
except Exception as e:
|
|
print(f"数据库初始化失败: {str(e)}")
|
|
raise
|
|
|
|
|
|
async def execute_migrations():
|
|
"""执行数据库迁移"""
|
|
try:
|
|
# 收集迁移文件
|
|
migration_files = []
|
|
for root, dirs, files in os.walk("apps"):
|
|
if "migrations" in dirs:
|
|
migration_path = os.path.join(root, "migrations")
|
|
migration_files.extend(glob.glob(os.path.join(migration_path, "migrations_*.py")))
|
|
|
|
# 按文件名排序
|
|
migration_files.sort()
|
|
|
|
for migration_file in migration_files:
|
|
file_name = os.path.basename(migration_file)
|
|
|
|
# 检查是否已执行
|
|
executed = await Tortoise.get_connection("default").execute_query(
|
|
"SELECT id FROM migrates WHERE migration_file = ?", [file_name]
|
|
)
|
|
|
|
if not executed[1]:
|
|
print(f"执行迁移: {file_name}")
|
|
# 导入并执行migration
|
|
module_path = migration_file.replace("/", ".").replace("\\", ".").replace(".py", "")
|
|
try:
|
|
migration_module = importlib.import_module(module_path)
|
|
if hasattr(migration_module, "migrate"):
|
|
await migration_module.migrate()
|
|
# 记录执行
|
|
await Tortoise.get_connection("default").execute_query(
|
|
"INSERT INTO migrates (migration_file) VALUES (?)",
|
|
[file_name]
|
|
)
|
|
print(f"迁移完成: {file_name}")
|
|
except Exception as e:
|
|
print(f"迁移 {file_name} 执行失败: {str(e)}")
|
|
raise
|
|
|
|
except Exception as e:
|
|
print(f"迁移过程发生错误: {str(e)}")
|
|
raise
|