python - SQLAlchemy DB 升级和 MySQL SSH

Question

我有一个 Flask 应用程序，需要在我的 config.py 文件中使用像这样的 SSHTunnel 连接到远程 MysqlDB，我在 init.py 文件中初始化:

sshtunnel.SSH_TIMEOUT = 5.0
sshtunnel.TUNNEL_TIMEOUT = 5.0

server =  sshtunnel.SSHTunnelForwarder(
    ('ssh.pythonanywhere.com', 22),
    ssh_password="mypassword",
    ssh_username="myusername",
    remote_bind_address=(myname.mysql.pythonanywhere-services.com', 3306))
server.start()

engine = create_engine('mysql+mysqldb://mynameb:dbpassword@127.0.0.1:%s/dbname' % server.local_bind_port)

连接似乎正常工作，但我无法从迁移中升级数据库(flask 数据库升级)，因为我没有使用 SQLALCHEMY_DATABASE_URI 连接到我的数据库。还有一种方法可以通过 ssh 连接到数据库来进行数据库升级吗？

Answer 1

Alembic 不必使用数据库 URI 连接到数据库即可运行升级。在您的 alembic\env.py 文件中将有一个如下所示的函数:

def run_migrations_online():
    """Run migrations in 'online' mode.

    In this scenario we need to create an Engine
    and associate a connection with the context.

    """
    connectable = engine_from_config(
        config.get_section(config.config_ini_section),
        prefix='sqlalchemy.',
        poolclass=pool.NullPool)

    with connectable.connect() as connection:
        context.configure(
            connection=connection,
            target_metadata=target_metadata
        )

        with context.begin_transaction():
            context.run_migrations()

重要的是，当 alembic 调用 connect() 时，变量 connectable 是一个 engine 实例。

因此你可以做这样的事情(这没有经过测试，但我自己做了类似的事情):

def run_migrations_online():
    """Run migrations in 'online' mode.

    In this scenario we need to create an Engine
    and associate a connection with the context.

    """

    sshtunnel.SSH_TIMEOUT = 5.0
    sshtunnel.TUNNEL_TIMEOUT = 5.0

    server =  sshtunnel.SSHTunnelForwarder(
        ('ssh.pythonanywhere.com', 22),
        ssh_password="mypassword",
        ssh_username="myusername",
        remote_bind_address=\
            (myname.mysql.pythonanywhere-services.com', 3306))
    server.start()

    connectable = create_engine(
        'mysql+mysqldb://mynameb:dbpassword@127.0.0.1:%s/dbname' % 
        server.local_bind_port
    )

    with connectable.connect() as connection:
        context.configure(
            connection=connection,
            target_metadata=target_metadata
        )

        with context.begin_transaction():
            context.run_migrations()

理想情况下，您可以将所有连接逻辑放在可以从 alembic/env.py 和项目中访问的地方，这样您只需定义一次，然后就可以导入engine 直接进入 env.py，但你明白了。