使用 sqlalchemy_utils ChoiceType 在 Flask 中迁移时出错答案

【问题标题】：Error at migrating in Flask with sqlalchemy_utils ChoiceType使用 sqlalchemy_utils ChoiceType 在 Flask 中迁移时出错
【发布时间】：2015-06-02 13:05:56
【问题描述】：

我有一个 Flask 模型：

class User(db.Model):
    ROLE_USER = 0
    ROLE_MODERATOR = 1
    ROLE_ADMIN = 2
    ROLES = [
        (ROLE_USER, u'Regular user'),
        (ROLE_MODERATOR, u'Moderator'),
        (ROLE_ADMIN, u'Admin')
    ]

    id = db.Column(db.Integer, primary_key = True)
    login = db.Column(db.String(32), nullable=False, unique=True)
    first_name = db.Column(db.String(32))
    last_name = db.Column(db.String(32))
    role = db.Column(ChoiceType(ROLES), nullable=False)

我已经用 flask-migrate 创建了一个迁移（db 是 Postgresql）：

def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.create_table('user',
    sa.Column('id', sa.Integer(), nullable=False),
    sa.Column('login', sa.String(length=32), nullable=False),
    sa.Column('first_name', sa.String(length=32), nullable=True),
    sa.Column('last_name', sa.String(length=32), nullable=True),
    sa.Column('role', sqlalchemy_utils.types.choice.ChoiceType(length=255), nullable=False),
    sa.PrimaryKeyConstraint('id'),
    sa.UniqueConstraint('login')
    )

迁移已成功创建，但是当我要升级时，出现此错误：

TypeError: <flask_script.commands.Command object at 0x7fada1e973d0>: __init__() got an unexpected keyword argument 'length'

谁能解释一下这个问题？

【问题讨论】：

标签： python postgresql flask sqlalchemy flask-migrate

【解决方案1】：

正如错误所说，ChoiceType 没有名为 length 的 init 参数：

http://sqlalchemy-utils.readthedocs.org/en/latest/data_types.html#module-sqlalchemy_utils.types.choice

你可以删除它并使用

sqlalchemy_utils.types.choice.ChoiceType(User.ROLES)

改为。

【讨论】：