python - Alembic API 如何获取原始 SQL 脚本

Question

我的要求是生成一个 SQL 脚本并将该脚本上传到云存储。我正在使用 Alembic 命令生成 SQL 脚本。

command.upgrade(alembic_cfg, revision='a:b', sql=True)

command.upgrade函数将 SQL 脚本打印到控制台。如果有多个修订版，那么就会有多个 INFO日志。
有没有办法将原始 SQL 脚本存储到变量中？

raw_sql = command.upgrade(alembic_cfg, revision='a:b', sql=True)
upload_script(raw_sql)

我知道可以使用 cli alembic upgrade head --sql > migration.sql

Answer 1

似乎 API 无法实现这一点。最后，我已经解决了 CLI 命令。

upgrade = subprocess.run(['alembic', 'upgrade', upgrade_revision_range, '--sql'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
raw_sql = upgrade.stdout.decode('utf-8')