nas-tools/dbscript_gen.py
2022-12-06 21:03:24 +08:00

13 lines
581 B
Python

import os
from config import Config
from alembic.config import Config as AlembicConfig
from alembic.command import revision as alembic_revision
db_version = input("请输入版本号:")
db_location = os.path.join(Config().get_config_path(), 'user.db').replace('\\', '/')
script_location = os.path.join(os.path.dirname(__file__), 'db_scripts').replace('\\', '/')
alembic_cfg = AlembicConfig()
alembic_cfg.set_main_option('script_location', script_location)
alembic_cfg.set_main_option('sqlalchemy.url', f"sqlite:///{db_location}")
alembic_revision(alembic_cfg, db_version, True)