12

Я разрабатываю веб-приложение, используя Flask под Python3. У меня проблема с типом postgresql enum на db migrate/upgrade.sqlalchemy postgresql enum не создает тип на db migrate

Я добавил столбец "статус" для модели:

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'] 

Почему миграция не создает тип «banner_status» «?

Что я делаю неправильно?

$ 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

От чтения http://stackoverflow.com/questions/14845203/altering-an-enum-field-using-alembic и https://bitbucket.org/zzzeek/alembic/issues/67/autogenerate-with-enums -on-postgres-dont, похоже, вам, возможно, придется вручную создавать и отбрасывать перечисления. –

ответ

25

Я решил эту проблему, используя это.

Я изменил код миграции и миграции выглядит следующим образом:

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 будет успешно выполнен.

+0

Для снижения рейтинга вы также можете избежать дублирования путем непосредственного выполнения SQL (по крайней мере, для postgres): 'op.execute (« DROP TYPE banner_status; »)' – Geekfish