def upgrade(): with op.batch_alter_table('upgrade_tasks') as batch_op: batch_op.add_column( sa.Column('status', sa.Enum(*constants.UPGRADE_STATUSES)))