Как узнать, является ли класс модели db или ndb

Я создал утилиту для обмена или архивации всех сущностей на вид. Но как узнать, является ли используемый 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

ОБНОВЛЕНИЕ: решение, которое я использовал. Смотрите также ответ Гвидо

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)

google-cloud-datastore google-app-engine app-engine-ndb
person voscausa    schedule 09.01.2013    source источник

Ответы (2)


arrow_upward
5
arrow_downward

Как насчет импорта 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.
person Guido van Rossum    schedule 10.01.2013

arrow_upward
5
arrow_downward

вы можете использовать атрибут, который существует только в ndb или наоборот.

например _has_repeated или _pre_get_hook, которые являются свойствами объектов ndb.
поэтому вы можете сделать:

self.ndb = hasattr(self, '_has_repeated')
person aschmid00    schedule 09.01.2013
comment
Да, это работает. Я уже пробовал что-то подобное, но вскоре сдался. Использование _pre_get_hook помогло. - person voscausa; 09.01.2013
comment
Спасибо за Ваш ответ. Это работает нормально. Если я не получу лучшего ответа, я приму его. Я жду, потому что _has_repeated не задокументировано как свойство. - person voscausa; 11.01.2013
comment
я бы пошел с ответом guidos ... не знаю, почему я не подумал об этом. - person aschmid00; 11.01.2013
comment
Это задокументированный ответ. Спасибо за вашу помощь. - person voscausa; 11.01.2013