Definiowanie kolumny wyliczenia SQLAlchemy z wyliczaniem w języku Python powoduje zwiększenie "ValueError: nie jest prawidłowym wyliczeniem".

Question

Próbuję podążać za this example, aby mieć kolumnę wyliczenia w tabeli korzystającej z typu Enum w języku Python. Definiuję enum, a następnie przekazuję go do kolumny, jak pokazano w przykładzie, ale otrzymuję ValueError: <enum 'FruitType'> is not a valid Enum. Jak poprawnie zdefiniować kolumnę wyliczenia SQLAlchemy z wyliczeniem 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 

Answer 1

typu kolumna powinna być sqlalchemy.types.Enum. Używasz ponownie języka Python Enum, który jest poprawny dla wartości, ale nie dla typu kolumny.

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