Question

在我的Flask app的postgres数据库上运行迁移时，我在Enum类型上遇到以下psycopg2错误：

INFO  [alembic.runtime.migration] Context impl PostgresqlImpl.
INFO  [alembic.runtime.migration] Will assume transactional DDL.
INFO  [alembic.runtime.migration] Running upgrade  -> 8753d3c9dbd1, empty message
Traceback (most recent call last):
  File "/home/jul/.miniconda3/envs/audiolabeling/lib/python3.6/site-packages/sqlalchemy/engine/base.py", line 1182, in _execute_context
    context)
  File "/home/jul/.miniconda3/envs/audiolabeling/lib/python3.6/site-packages/sqlalchemy/engine/default.py", line 470, in do_execute
    cursor.execute(statement, parameters)
psycopg2.ProgrammingError: type "feedbacktype" already exists

我尝试删除了alembic_version表和迁移文件夹，并重新运行了db init / migrate / upgrade，但没有成功。

我不得不完全放弃数据库以解决问题。

有没有办法在它们已经存在时不重新创建类型？

models.py

from audiolabeling import db
import enum


class FeedBackType(enum.Enum):
    NONE = "none"
    HIDDENIMAGE = "hiddenImage"

class Task(db.Model):

    id = db.Column(db.Integer, primary_key=True)
    feedback = db.Column(db.Enum(FeedBackType))

    def __repr__(self):
        return '<id {}>'.format(self.id)

Answer 1

您可以使用feedback = db.Column(db.Enum(["NONE", "HIDDENIMAGE"]))替换您的代码。

Answer 2

编辑：以下内容不适用于Alembic。同样的问题在这里：https://github.com/sqlalchemy/alembic/issues/566

在创建列时，使用sqlalchemy.dialects.postgresql.ENUM代替db.Enum。

ENUM具有可选参数create_type，应将其设置为False。

像这样：

from sqlalchemy.dialects.postgresql import ENUM

...

class Task(db.Model):

    id = db.Column(db.Integer, primary_key=True)
    feedback = db.Column(ENUM(FeedBackType, create_type=False))

def __repr__(self):
    return '<id {}>'.format(self.id)

Answer 3

答案可能已经很晚了，但仍会尝试将其发布给将来寻找此问题的其他人。（和我一样）

还有一个 create_type kwarg 表示 postgresql.ENUM。用于在 alembic 迁移脚本中设置枚举类型列的架构。

这是我的新列定义的样子。（使用现有的枚举）


sa.Column('ActionType', postgresql.ENUM('Primary', 'Secondary', name='actiontype', create_type=False), nullable=True),

现在将使用现有枚举作为新列，而不创建新列。

我在我的需求文件中使用了 SQLAlchemy==1.1.1 和 alembic==0.8.8。

在Flask（数据库）数据库迁移中的枚举类型上的psycopg2.ProgrammingError

3 个答案: