Как откатить все миграции, если произошла ошибка?

Question

Кажется, что есть способ откат всех изменений при ошибке, но пример сценария кажется странным, потому что ссылка session, используемая в сценарии, никогда не определяется и не импортируется:

from alembic import context
import myapp
import sys

db_1 = myapp.db_1
db_2 = myapp.db_2

def run_migrations_offline():
    """Run migrations *without* a SQL connection."""

    for name, engine, file_ in [
        ("db1", db_1, "db1.sql"),
        ("db2", db_2, "db2.sql"),
    ]:
        context.configure(
                    url=engine.url,
                    transactional_ddl=False,
                    output_buffer=open(file_, 'w'))
        context.execute("-- running migrations for '%s'" % name)
        context.run_migrations(name=name)
        sys.stderr.write("Wrote file '%s'" % file_)

def run_migrations_online():
    """Run migrations *with* a SQL connection."""

    for name, engine in [
        ("db1", db_1),
        ("db2", db_2),
    ]:
        connection = engine.connect()
        context.configure(connection=connection)
        try:
            context.run_migrations(name=name)
            session.commit()    # <---
        except:
            session.rollback()  # <---
            raise

if context.is_offline_mode():
    run_migrations_offline()
else:
    run_migrations_online()

Это очень странно (imho), что это не поведение по умолчанию, но я хочу откатить каждые изменения из сценария миграции в случае возникновения ошибки.

Как я могу это сделать?