如何在事务中包装create和sqlalchemy-migrate alter语句

Question

我正在尝试使用sqlalchemy迁移和sqlalchemy- table.create table.rename方法以及一些insert into select语句对数据库进行一些数据库模式更改。我想把所有这些都包装在一个事务中。我想不出该怎么做。这是我尝试过的：

engine = sqlalchemy.engine_from_config(conf.local_conf, 'sqlalchemy.')
trans = engine.connect().begin()
try:
    old_metatadata.tables['address'].rename('address_migrate_tmp', connection=trans)
    new_metatadata.tables['address'].create(connection=trans)
except:
    trans.rollback()
    raise
else:
    trans.commit()

但是它的错误是：

AttributeError: 'RootTransaction' object has no attribute '_run_visitor'

(我尝试过使用sqlalchemy-migrate column.alter(column.alter=‘newname’)，但这是错误的，并且在事务中不起作用，因此使我的数据库处于损坏状态。我还需要重命名多个列，因此我决定使用自己的代码。)

Answer 1

啊-我只需要使用在其上创建事务的连接。

engine = sqlalchemy.engine_from_config(conf.local_conf, 'sqlalchemy.')
conn = engine.connect()
trans = conn.begin()
try:
    old_metatadata.tables['address'].rename('address_migrate_tmp', connection=conn)
    new_metatadata.tables['address'].create(bind=conn)
except:
    trans.rollback()
    raise
else:
    trans.commit()