2016-03-21 7 views
13

Ich versuche, this example zu folgen, um eine Enum-Spalte in einer Tabelle, die Python Enum Typ verwendet. Ich definiere die Enumeration und gebe sie dann an die Spalte, wie im Beispiel gezeigt, aber ich bekomme ValueError: <enum 'FruitType'> is not a valid Enum. Wie definiere ich eine SQLAlchemy enum-Spalte mit einer Python-Enumeration richtig?Definieren von SQLAlchemy Enum-Spalte mit Python Enum löst "ValueError: keine gültige enum"

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 

Antwort

23

Der Spaltentyp sqlalchemy.types.Enum sein sollte. Sie verwenden wieder den Python-Typ Enum, der für den Wert, nicht aber für den Spaltentyp gilt.

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

Dies funktioniert mit SQLAlchemy 1.1 und höher – Anconia