如何在SQLAlchemy和Alembic中使用Enum？

Question

我的帖子：

class Post(Base):
    __tablename__ = 'posts'

    title = db.Column(db.String(120), nullable=False)
    description = db.Column(db.String(2048), nullable=False)

我想添加枚举status。所以，我创建了一个新的枚举：

^{pr2}$

并在模型中添加了一个新字段：

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

完成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))

在尝试升级数据库后，我得到：

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' NOT NULL"]

1. 为什么类型poststatus没有在数据库级别自动创建？在类似的迁移过程中也是如此。在
2. 如何正确指定server_default选项？我需要ORM级别的默认值和DB级别的默认值，因为我正在更改现有行，所以ORM默认值不适用。在
3. 为什么DB中的实际值是'DRAFT'、'APPROVE'、'PUBLISHED'，而不是{}等等？我想应该有枚举值，而不是名称。在

提前谢谢你。在

Answer 1

来自官方文件：https://docs.python.org/3/library/enum.html#creating-an-enum

import enum

class PostStatus(enum.Enum):
    DRAFT = 0
    APPROVE = 1
    PUBLISHED = 2

据此：

^{pr2}$

1）PostStatus不是DB模型，它只是一个包含状态ID的类

2）没事

3）你不必在数据库中存储状态字符串，你最好改用id