python - 如何通过代码使用 Python Alembic 运行迁移？

Question

我正在尝试使用 sqlAlchemy 和 Alembic 迁移 SQL 数据库。我想直接在 Python 代码中制作简单的迁移脚本，而不是使用 Alembic CLI，如文档中所述。

我只在这个主题上找到了这个 SO 问题:Using Alembic API from inside application code

使用这个问题 + Flask-Alembic 源代码，我尝试了这些简单的命令。引擎链接到我的数据库，元数据包含迁移信息。

我觉得我已经很接近了，解决方案应该在一行代码中......我很挣扎。

from alembic.config import Config
from alembic import command, autogenerate
from alembic.script import ScriptDirectory
from alembic.runtime.environment import EnvironmentContext

alembic_cfg = Config()
alembic_cfg.set_main_option("script_location", "migrations")
alembic_cfg.set_main_option("url", "postgresql://user:pass@postgres:5432/mydb")

alembic_script = ScriptDirectory.from_config(alembic_cfg)
alembic_env = EnvironmentContext(alembic_cfg, alembic_script)

conn = engine.connect()
alembic_env.configure(connection=conn, target_metadata=metadata)
alembic_context = alembic_env.get_context()

我能够使用以下命令来查看它是否有效并检测哪些字段必须迁移:

autogenerate.compare_metadata(alembic_context, metadata)
autogenerate.produce_migrations(alembic_context, metadata)

但是，我无法运行迁移。我尝试了几个命令，但总是出错...

例如，如果我运行:

with alembic_env.begin_transaction():
    alembic_env.run_migrations()

我明白了:

/usr/local/lib/python2.7/site-packages/alembic/runtime/migration.pyc in run_migrations(self, **kw)
    301         head_maintainer = HeadMaintainer(self, heads)
    302 
--> 303         for step in self._migrations_fn(heads, self):
    304             with self.begin_transaction(_per_migration=True):
    305                 if self.as_sql and not head_maintainer.heads:

TypeError: 'NoneType' object is not callable

谁能帮帮我。我认为解决方案是单行的，但我找不到如何在我的数据库上运行迁移...

有关信息，我从未使用 Alembic 在此数据库上进行过任何迁移，也许需要“初始化”？

非常感谢。

Answer 1

self._migrations_fn(heads, self) 中的步骤:

此处我们将描述您要迁移的顺序。该序列由步骤组成，每个步骤都有 step.migration_fn(**kw)。

如何修复

您需要的最后一步是在执行 alembic_env.configure 时添加 _migrations_fn。

def do_upgrade(revision, context):
    return alembic_script._upgrade_revs(script.get_heads(), revision)

alembic_env.configure(connection=conn, target_metadata=metadata, fn=do_upgrade)

script.get_heads() 返回最后一次迁移。替换，如果您需要特定的修订，而不是最后一个。

相关链接:

• Using Alembic API from inside application code

  https://bitbucket.org/davidism/flask-alembic/src/e036f063af1ba667c74560264639a93d812dfd51/flask_alembic/extension.py?at=default&fileviewer=file-view-default#extension.py-324