如何在Alembic迁移中忽略Postgres角色不存在的失败？

你的方案完全可行，核心问题在于不需要手动调用op.rollback()——Alembic默认会在迁移操作抛出异常时自动回滚当前事务，强行手动回滚反而可能导致事务状态混乱。只需在捕获到「角色不存在」的特定异常时，跳过操作即可让迁移流程正常继续。

修正后的代码如下：

from psycopg.errors import UndefinedObject
from sqlalchemy.exc import ProgrammingError

def upgrade():
    try:
        op.execute("alter role pgsqladmin valid until '9999-12-31 12:00:00 +0'")
    except ProgrammingError as e:
        if isinstance(e.__context__, UndefinedObject):
            print("角色pgsqladmin不存在，跳过该操作")
        else:
            # 其他类型的编程错误（比如SQL语法错误），需要重新抛出以终止迁移
            raise

为什么不需要op.rollback()？ #

Alembic执行单个迁移脚本时，默认会把整个脚本逻辑包装在一个事务中。当op.execute抛出异常时，Alembic会自动触发事务回滚，不需要开发者手动干预。如果此时调用op.rollback()，反而会干扰Alembic自身的事务管理流程。

进阶：手动管理事务的场景（可选） #

如果你的迁移脚本包含多个独立操作，需要更精细的事务控制，可以通过op.get_bind()获取数据库连接，手动管理事务生命周期：

from psycopg.errors import UndefinedObject
from sqlalchemy.exc import ProgrammingError

def upgrade():
    conn = op.get_bind()
    try:
        conn.execute("alter role pgsqladmin valid until '9999-12-31 12:00:00 +0'")
    except ProgrammingError as e:
        if isinstance(e.__context__, UndefinedObject):
            print("角色pgsqladmin不存在，跳过该操作")
            conn.rollback()
        else:
            conn.rollback()
            raise
    else:
        conn.commit()

不过对于单个角色修改操作来说，第一种简洁方案完全足够满足需求。

内容的提问来源于stack exchange，提问作者Guillaume