google-app-engine - 如何确定模型类是 db 还是 ndb

Question

我创建了一个实用程序来交换或压缩所有实体的一种。但是我怎样才能知道使用的 model_class 是 db.Model 还是 ndb.Model？

def _encode_entity(self, entity):                                             

    if self.ndb :
        entity_dict = entity.to_dict()                                                     
        self.entity_eid = entity.key.id()
        entity_dict['NDB'] = True
    else :
        entity_dict = db.to_dict(entity)
        self.entity_eid = entity.key().name()
        entity_dict['NDB'] = False

    ....

现在我使用：

def queryKind(self):

    try :
        self.query = self.model_class.query()
        self.ndb = True
    except AttributeError :
        self.query = self.model_class.all()
        self.ndb = False
    return self.make(self._encode_entity)       # make a zip or a page

更新：我使用的解决方案。另见Guido的回答

self.kind = 'Greeting'
module = __import__('models', globals(), locals(), [self.kind], -1)
self.model_class = getattr(module, self.kind)
entity = self.model_class()

if isinstance(entity, ndb.Model): 
    self.ndb = True
    self.query = self.model_class.query()
elif isinstance(entity, db.Model): 
    self.ndb = False
    self.query = self.model_class.all()
else :
    raise ValueError('Failed to classify entities of kind : ' + self.kind)

Answer 1

您可以使用仅存在于ndb或以其他方式存在的属性。

例如_has_repeated或_pre_get_hook哪些是ndb实体的属性。
所以你可以这样做：

self.ndb = hasattr(self, '_has_repeated')

Answer 2

如何导入 ndb 和 db，并测试作为各自模型类实例的实体？

if isinstance(entity, ndb.Model):
    # Do it the NDB way.
elif isinstance(entity, db.Model):
    # Do it the db way.
else:
    # Fail.  Not an entity.