將枚舉傳遞給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以外的任何值。