相关疑难解决方法(0)

如何在SQLAlchemy和Alembic中使用Enum?

这是我的Post模型:

class Post(Base):
    __tablename__ = 'posts'

    title = db.Column(db.String(120), nullable=False)
    description = db.Column(db.String(2048), nullable=False)
Run Code Online (Sandbox Code Playgroud)

我想在其中添加枚举status。因此,我创建了一个新的Enum:

import enum

class PostStatus(enum.Enum):
    DRAFT='draft'
    APPROVE='approve'
    PUBLISHED='published'
Run Code Online (Sandbox Code Playgroud)

并为模型添加了一个新字段:

class Post(Base):
    ...
    status = db.Column(db.Enum(PostStatus), nullable=False, default=PostStatus.DRAFT.value, server_default=PostStatus.DRAFT.value)
Run Code Online (Sandbox Code Playgroud)

完成后FLASK_APP=server.py flask db migrate,生成了这样的迁移:

def upgrade():
    op.add_column('posts', sa.Column('status', sa.Enum('DRAFT', 'APPROVE', 'PUBLISHED', name='poststatus'), server_default='draft', nullable=False))
Run Code Online (Sandbox Code Playgroud)

尝试升级数据库后,我得到:

sqlalchemy.exc.ProgrammingError: (psycopg2.ProgrammingError) type "poststatus" does not exist
LINE 1: ALTER TABLE posts ADD COLUMN status poststatus DEFAULT 'draf...
                                            ^
 [SQL: "ALTER TABLE posts ADD COLUMN status poststatus DEFAULT 'draft' …
Run Code Online (Sandbox Code Playgroud)

python sqlalchemy flask-sqlalchemy alembic

6
推荐指数
4
解决办法
3746
查看次数

标签 统计

alembic ×1

flask-sqlalchemy ×1

python ×1

sqlalchemy ×1