20 lines
711 B
Python
20 lines
711 B
Python
import importlib
|
|
from pathlib import Path
|
|
|
|
from alembic.config import Config as AlembicConfig
|
|
from alembic.command import revision as alembic_revision
|
|
|
|
from app.core.config import settings
|
|
|
|
# 导入模块，避免建表缺失
|
|
for module in Path(__file__).with_name("models").glob("*.py"):
|
|
importlib.import_module(f"app.db.models.{module.stem}")
|
|
|
|
db_version = input("请输入版本号：")
|
|
db_location = settings.CONFIG_PATH / 'user.db'
|
|
script_location = settings.ROOT_PATH / 'alembic'
|
|
alembic_cfg = AlembicConfig()
|
|
alembic_cfg.set_main_option('script_location', str(script_location))
|
|
alembic_cfg.set_main_option('sqlalchemy.url', f"sqlite:///{db_location}")
|
|
alembic_revision(alembic_cfg, db_version, True)
|