Alembic迁移遇InvalidSchemaError：枚举创建架构选择异常排查

问题背景 #

使用Alembic创建带枚举字段的表时，初始代码会把枚举默认创建在PostgreSQL的public架构中，这在允许访问public的环境中能正常运行，但在限制public架构访问的私有环境里会触发InvalidSchemaError No schema selected to create in错误。修改后的迁移代码在本地测试正常，但私有环境仍报错，注释枚举相关代码后迁移可正常执行。

解决方案 #

核心是明确指定枚举的目标架构，确保枚举类型创建在自定义的my_personal_schema中，而非默认的public。以下是修正后的完整迁移代码：

from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

def upgrade():
    # 先创建不含枚举列的表
    op.create_table(
        'test_table',
        sa.Column('name', sa.String(length=250), nullable=False),
        sa.Column('id', sa.BigInteger(), autoincrement=True, nullable=False),
        sa.Column('is_deleted', sa.Boolean(), server_default=sa.text('false'), nullable=False),
        sa.PrimaryKeyConstraint('id', name=op.f('pk_test_table')),
        schema='my_personal_schema'
    )

    # 1. 创建指定架构的枚举类型
    status_enum = postgresql.ENUM(
        'CREATED', 'COMPLETED',
        name='status',
        schema='my_personal_schema',  # 强制枚举创建在目标架构
        create_type=True
    )
    # 绑定连接创建枚举，checkfirst避免重复创建
    status_enum.create(op.get_bind(), checkfirst=True)

    # 2. 给表添加枚举列
    with op.batch_alter_table('test_table', schema='my_personal_schema') as batch_op:
        batch_op.add_column(
            sa.Column(
                'status',
                status_enum,  # 使用已创建的带架构的枚举类型
                nullable=False
            )
        )

def downgrade():
    # 先删除枚举列
    with op.batch_alter_table('test_table', schema='my_personal_schema') as batch_op:
        batch_op.drop_column('status')

    # 删除目标架构下的枚举类型
    status_enum = postgresql.ENUM(
        'CREATED', 'COMPLETED',
        name='status',
        schema='my_personal_schema'
    )
    status_enum.drop(op.get_bind(), checkfirst=True)

    # 最后删除表
    op.drop_table('test_table', schema='my_personal_schema')

关键修正点说明 #

1. 枚举指定schema参数：创建postgresql.ENUM时添加schema='my_personal_schema'，强制PostgreSQL将枚举类型创建在目标架构中，彻底摆脱对public架构的依赖。
2. 独立创建枚举：直接通过op.get_bind()创建枚举，确保上下文使用目标架构的权限，避免batch_op.get_bind()可能存在的schema上下文继承问题。
3. 完整降级逻辑：降级时先删除枚举列再删除枚举类型，避免因依赖关系导致删除失败。

旧代码问题分析 #

之前的修改中，postgresql.ENUM未指定schema参数，即便设置了create_type=False，当枚举不存在时，SQLAlchemy仍会尝试自动创建枚举类型，但此时没有指定schema，且数据库用户无public访问权限，因此触发InvalidSchemaError。

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