在我的Flask应用的postgres数据库上运行迁移时,我在枚举类型上收到以下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表和migrations文件夹,然后重新运行flask 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)
您可以使用feedback = db.Column(db.Enum(["NONE", "HIDDENIMAGE"]))
替换您的代码。
编辑:以下内容不适用于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)