将枚举传递给python中的ndb.Model字段

Question

我发现How can I represent an 'Enum' in Python?是关于如何在python中创建枚举的。我在我的ndb.Model中有一个字段，我想接受其中一个枚举值。我是否只需将字段设置为StringProperty？我的枚举是

def enum(**enums): 
    return type('Enum',(), enums) 

ALPHA = enum(A="A", B="B", C="C", D="D") 

Answer 1

这在ProtoRPC Python API完全支持，它不值得滚动自己。

一个简单枚举看起来像下面这样：

from protorpc import messages 

class Alpha(messages.Enum): 
    A = 0 
    B = 1 
    C = 2 
    D = 3 

事实证明，ndb具有用于存储protorpc对象msgprop模块，这是documented。

所以来存储您的Alpha枚举，你会做到以下几点：

from google.appengine.ext import ndb 
from google.appengine.ext.ndb import msgprop 

class Part(ndb.Model): 
    alpha = msgprop.EnumProperty(Alpha, required=True) 
    ... 

编辑：由于hadware指出，一个msgprop.EnumProperty默认情况下未编入索引。如果要执行过这样的属性查询你需要定义属性为

alpha = msgprop.EnumProperty(Alpha, required=True, indexed=True) 

，然后执行查询

ndb.query(Part.alpha == Alpha.B) 

或使用比Alpha.B以外的任何值。