如何使用Alembic将ENUM类型列转换为INTEGER类型外键列？

问题根源 #

PostgreSQL无法直接将ENUM类型强制转换为INTEGER——你的ENUM成员是字符串形式（如'6'），直接用::integer转换会触发类型转换失败。同时你当前迁移文件里的postgresql_using="NULL"是错误配置，这会导致转换逻辑完全失效。

修正后的迁移代码 #

def upgrade() -> None:
    # 先将ENUM列转为INTEGER，正确的转换逻辑：ENUM -> TEXT -> INTEGER
    op.alter_column(
        'aluno', 
        'serie_aluno',
        existing_type=postgresql.ENUM('6', '7', '8', '9', name='serie_enum'),
        type_=sa.Integer(),
        existing_nullable=False,
        # 关键：先转文本再转整数，匹配ENUM的字符串值到整数
        postgresql_using="serie_aluno::text::integer"
    )
    # 显式指定外键名称，避免后续降级时找不到约束
    op.create_foreign_key(
        'fk_aluno_serie',
        'aluno', 
        'serie', 
        ['serie_aluno'], 
        ['id_serie']
    )

def downgrade() -> None:
    # 用指定的名称删除外键
    op.drop_constraint('fk_aluno_serie', 'aluno', type_='foreignkey')
    # 将INTEGER转回ENUM，同样需要先转文本再匹配ENUM成员
    op.alter_column(
        'aluno', 
        'serie_aluno',
        existing_type=sa.Integer(),
        type_=postgresql.ENUM('6', '7', '8', '9', name='serie_enum'),
        existing_nullable=False,
        postgresql_using="serie_aluno::text::serie_enum"
    )

关键说明 #

1. 类型转换逻辑：

   • 升级时：serie_aluno::text::integer 先把ENUM的字符串值（如'6'）转为文本类型，再转为对应的整数6，这是PostgreSQL支持的合法转换路径。
   • 降级时：serie_aluno::text::serie_enum 把整数转为文本，再匹配回原ENUM的成员值，确保数据类型可逆。

2. 外键约束名称：
   不要用None自动生成约束名，显式指定名称（如fk_aluno_serie）可以避免降级时因约束名不明确导致的删除失败。

3. 数据一致性检查：
   确保serie表的id_serie列的值与原ENUM成员转成整数后完全一致（比如id_serie的值是6、7、8、9），否则创建外键时会因数据不匹配触发约束冲突错误。

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