Ошибка при переносе и обновлении во Flask с помощью sqlalchemy_utils ChoiceType - PullRequest
0 голосов
/ 17 мая 2018

Я использую flask-migrate для настройки изменений в моей базе данных.

$python manage.py db init
$python manage.py db migrate

Они работают нормально, но когда я использую

$python manage.py db upgrade

Эта ошибка возникает

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

Это мой models.py

class JobsModel(db.Model):
JOB_STATUS_TYPES = [
  (u"data_upload_in_progress", u"Data Upload in Progress"),
  (u"data_upload_success", u"Data Upload Success"),
  (u"data_validation_success", u"Data Validation Success"),
  (u"data_validation_error", u"Data Validation Error")
   ]

 __tablename__ = 'jobs'

 job_id = db.Column(db.Integer, primary_key = True)
 usecase_id = db.Column(db.Integer, db.ForeignKey('use_cases.id'), nullable = False)
 usecase = db.relationship('UseCaseModel')
 user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
 version_id = db.Column(db.Float, nullable=True)
 job_status = db.Column(ChoiceType(JOB_STATUS_TYPES))
 dataset_id = db.Column(db.Integer, db.ForeignKey('datasets.dataset_id'), nullable=False)
 dataset = db.relationship('DataSetModel')
 training_details = db.Column(JSON)
 testing_details = db.Column(JSON)

Это мой файл после выполнения команды переноса 2easaw23523sa.py

from alembic import op
import sqlalchemy as sa
from mi import mig
def upgrade():
   op.create_table('jobs',
   sa.Column('job_id', sa.Integer(), nullable=False),
   sa.Column('usecase_id', sa.Integer(), nullable=False),
   sa.Column('user_id', sa.Integer(), nullable=False),
   sa.Column('version_id', sa.Float(), nullable=True),
   sa.Column('job_status', mig.ChoiceType(length=255), nullable=True),
   sa.Column('dataset_id', sa.Integer(), nullable=False),
   sa.Column('training_details', sa.JSON(), nullable=True),
   sa.Column('testing_details', sa.JSON(), nullable=True),
   sa.ForeignKeyConstraint(['dataset_id'], ['datasets.dataset_id'], ),
   sa.ForeignKeyConstraint(['usecase_id'], ['use_cases.id'], ),
   sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
   sa.PrimaryKeyConstraint('job_id')
   )

def downgrade():
   # ### commands auto generated by Alembic - please adjust! ###
   op.drop_table('jobs')

# ### end Alembic commands ###

Примечание - выписка из mi import mig появляется потому, что при использовании процесса, указанного в этом ответе

1 Ответ

0 голосов
/ 31 мая 2018

Это можно исправить, заменив length=255 на {1:2} (или любой другой объект python, который является Truthy и может быть передан в dict() без исключения).

т.е. заменить строку

sa.Column('job_status', mig.ChoiceType(length=255), nullable=True),

С

sa.Column('job_status', mig.ChoiceType({1:2}), nullable=True),

...