from alembic import config, script
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

# 数据库连接
engine = create_engine('sqlite:///example.db')
Session = sessionmaker(bind=engine)
session = Session()

# 运行迁移
config = config.Config('alembic.ini')
script_directory = script.ScriptDirectory.from_config(config)
with engine.connect() as connection:
    context = script_directory.run_migrations(connection=connection)

print('Database migration completed.')