首页 文章

sqlalchemy postgresql枚举不会在db migrate上创建类型

提问于
浏览
16

我在Python3下使用Flask开发了一个web应用程序 . 我在db migrate / upgrade上遇到postgresql枚举类型的问题 .

我在模型中添加了一个“状态”列:

class Banner(db.Model):
    ...
    status = db.Column(db.Enum('active', 'inactive', 'archive', name='banner_status'))
    ...

python manage.py db migrate 生成的迁移是:

from alembic import op
import sqlalchemy as sa

def upgrade():
    op.add_column('banner', sa.Column('status', sa.Enum('active', 'inactive', 'archive', name='banner_status'), nullable=True))

def downgrade():
    op.drop_column('banner', 'status')

当我做 python manage.py db upgrade 时,我收到一个错误:

...
sqlalchemy.exc.ProgrammingError: (psycopg2.ProgrammingError) type "banner_status" does not exist
LINE 1: ALTER TABLE banner ADD COLUMN status banner_status

 [SQL: 'ALTER TABLE banner ADD COLUMN status banner_status']

Why migration does not create a type "banner_status"?

What am I doing wrong?

$ pip freeze
alembic==0.8.6
Flask==0.10.1
Flask-Fixtures==0.3.3
Flask-Login==0.3.2
Flask-Migrate==1.8.0
Flask-Script==2.0.5
Flask-SQLAlchemy==2.1
itsdangerous==0.24
Jinja2==2.8
Mako==1.0.4
MarkupSafe==0.23
psycopg2==2.6.1
python-editor==1.0
requests==2.10.0
SQLAlchemy==1.0.13
Werkzeug==0.11.9

1 回答

  • 34

    我用这个决定了这个问题 .

    我更改了迁移代码,迁移如下所示:

    from alembic import op
    import sqlalchemy as sa
    from sqlalchemy.dialects import postgresql
    
    def upgrade():
        banner_status = postgresql.ENUM('active', 'inactive', 'archive', name='banner_status')
        banner_status.create(op.get_bind())
    
        op.add_column('banner', sa.Column('status', sa.Enum('active', 'inactive', 'archive', name='banner_status'), nullable=True))
    
    def downgrade():
        op.drop_column('banner', 'status')
    
        banner_status = postgresql.ENUM('active', 'inactive', 'archive', name='banner_status')
        banner_status.drop(op.get_bind())
    

    现在 python manage.py db upgrade\downgrade 已成功执行 .

相关问题