dbscript_gen.py

import os
from config import Config
from alembic.config import Config as AlembicConfig
from alembic.command import revision as alembic_revision
from alembic.command import upgrade as alembic_upgrade

db_version = input("请输入版本号：")
db_location = os.path.join(Config().get_config_path(), 'user.db').replace('\\', '/')
script_location = os.path.join(os.path.dirname(__file__), '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_upgrade(alembic_cfg, "head")
alembic_revision(alembic_cfg, db_version, True)