Подтвердить что ты не робот

Определение столбца перечисления SQLAlchemy с перечислением Python вызывает "ValueError: недействительное перечисление"

Я пытаюсь следовать этому примеру, чтобы иметь столбец перечисления в таблице, использующей тип Python Enum. Я определяю перечисление, а затем передаю его в столбец, как показано в примере, но я получаю ValueError: <enum 'FruitType'> is not a valid Enum. Как правильно определить столбец перечисления SQLAlchemy с перечислением Python?

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
import enum

app = Flask(__name__)
db = SQLAlchemy(app)

class FruitType(enum.Enum):
    APPLE = "Crunchy apple"
    BANANA = "Sweet banana"

class MyTable(db.Model):
    id = db.Column(db.Integer, primary_key = True)
    fruit_type = db.Column(enum.Enum(FruitType))
File "why.py", line 32, in <module>
    class MyTable(db.Model):
  File "why.py", line 34, in MyTable
    fruit_type = db.Column(enum.Enum(FruitType))
  File "/usr/lib/python2.7/dist-packages/enum/__init__.py", line 330, in __call__
    return cls.__new__(cls, value)
  File "/usr/lib/python2.7/dist-packages/enum/__init__.py", line 642, in __new__
    raise ValueError("%s is not a valid %s" % (value, cls.__name__))
ValueError: <enum 'FruitType'> is not a valid Enum
4b9b3361

Ответ 1

Тип столбца должен быть sqlalchemy.types.Enum. Вы снова используете тип Python Enum, который действителен для значения, но не тип столбца.

class MyTable(db.Model):
    id = db.Column(db.Integer, primary_key = True)
    fruit_type = db.Column(db.Enum(FruitType))