Question

使用SQLAlchemy和Alembic更改现有的postgresql.ENUM列时遇到麻烦。

我想向agrebic中的postgresql.ENUM类型列添加/删除一个值。

具体来说，当前的枚举类型是由以下两个Alembic版本创建的：

# revision 1
def upgrade():
    op.create_table('kernels',
        sa.Column('status', sa.String(), nullable=True),
        ...
    )


# revision 2
kernelstatus_choices = (
    'PREPARING', 'BUILDING', 'RUNNING',
    'RESTARTING', 'RESIZING', 'SUSPENDED',
    'TERMINATING', 'TERMINATED', 'ERROR',
)
kernelstatus = postgresql.ENUM(
    *kernelstatus_choices,
    name='kernelstatus')

def upgrade():
    op.alter_column('kernels', column_name='status', 
                    type_=sa.Enum(*kernelstatus_choices, name='kernelstatus'),
                    postgresql_using='status::kernelstatus')

现在，我想将'PENDING'状态添加到kernelstatus类型。因此，我通过引用some articles来实现如下。

prev_kernelstatus_choices = (
    'PREPARING', 'BUILDING', 'RUNNING',
    'RESTARTING', 'RESIZING', 'SUSPENDED',
    'TERMINATING', 'TERMINATED', 'ERROR',
)

prev_kernelstatus = postgresql.ENUM(
    *prev_kernelstatus_choices,
    name='kernelstatus')

curr_kernelstatus_choices = ('PENDING',) + prev_kernelstatus_choices

curr_kernelstatus = postgresql.ENUM(
    *curr_kernelstatus_choices,
    name='kernelstatus')

def upgrade():
    op.execute('ALTER TYPE kernelstatus RENAME TO kernelstatus_old;')
    curr_kernelstatus.create(op.get_bind())
    op.alter_column('kernels', column_name='status', 
                    type_=sa.Enum(*curr_kernelstatus_choices, name='kernelstatus'),
                    postgresql_using='status::text::kernelstatus')
    op.execute('DROP TYPE kernelstatus_old;')

但是它不断产生以下错误：

sqlalchemy.exc.ProgrammingError: (psycopg2.ProgrammingError) operator does not exist: kernelstatus <> kernelstatus_old
HINT:  No operator matches the given name and argument type(s). You might need to add explicit type casts.
[SQL: 'ALTER TABLE kernels ALTER COLUMN status TYPE kernelstatus USING status::text::kernelstatus']

我已经尝试过使用adding value to enum type的解决方案，但这不适用于Alembic，因为每个Alembic修订版都在事务中运行，而ALTER TYPE语句不能在事务中运行。另外，应该有一个downgrade()的代码，并且没有在PostgreSQL中从枚举类型中删除值的语句，因此对我而言，仅将值添加到枚举类型中并不是最终的解决方案。

有人可以帮我吗？

Answer 1

我弄清了这种情况的根本原因（我已经与提问者进行了合作）：由于kernels表具有唯一的约束，可以检查（sess_id列的唯一性，因此枚举类型为status的枚举类型kernelstatus的{{1}}列。

要转换'TERMINATED'列的枚举类型，需要显式定义status SQL运算符，以使此约束检查在整个转换过程中保持一致。问题在于PostgreSQL的错误消息只是说“运算符不存在”，而不是“必须定义运算符以在转换期间保持约束”。

因此，我可以编写工作迁移，方法是临时添加一个比较运算符，该运算符通过将旧的/新的枚举值都转换为字符串来实现：

<>

添加/删除alembic中的PostgreSQL ENUM类型的值

1 个答案: