3

如何为 EndpointsModel 设置父/祖先并让数据存储区自动生成实体 ID/密钥?

我尝试调整keys_with_ancestors示例,但似乎遇到了一些障碍,因为它需要同时指定 id 和 parent。我想做类似的事情,除了只提供父 ID 或密钥,并让应用引擎数据存储区自动生成实体 ID/密钥。

以下显示了我将如何仅使用 NDB 来完成它。

class Parent(ndb.Model):
    name = ndb.StringProperty()

class MyModel(ndb.Model):
    attr1 = ndb.StringProperty()
    attr2 = ndb.StringProperty()

p = Parent(name="Jerry")
p_key = p.put()  # or retrieve the key from somewhere else

mod = MyModel(parent=p_key)
mod.put()

这是可能的吗?有人能指出我正确的方向吗?谢谢。

4

1 回答 1

5

keys_with_ancestors示例之后,假设我们有相同的导入,并且已经MyParent以相同的方式定义了类。

TL;DR 的答案本质上是传递parent=给模型构造函数等同于创建一个键,None并将其作为种类 ID 对列表中的最后一个 ID。例如,对于一个类MyModel

>>> parent = ndb.Key(MyModel, 1)
>>> child = MyModel(parent=parent)
>>> print child.key
ndb.Key('MyModel', 1, 'MyModel', None)

为了对示例执行此操作,我们可以简单地忽略id

class MyModel(EndpointsModel):
  _parent = None

  attr1 = ndb.StringProperty()
  attr2 = ndb.StringProperty()
  created = ndb.DateTimeProperty(auto_now_add=True)

并在设置器中简单地设置半生不熟的密钥,不要尝试从数据存储中检索(因为密钥不完整):

  def ParentSet(self, value):
    if not isinstance(value, basestring):
      raise TypeError('Parent name must be a string.')

    self._parent = value
    if ndb.Key(MyParent, value).get() is None:
      raise endpoints.NotFoundException('Parent %s does not exist.' % value)
    self.key = ndb.Key(MyParent, self._parent, MyModel, None)

    self._endpoints_query_info.ancestor = ndb.Key(MyParent, value)

同样,在 getter 中,您可以直接从键中检索父级(尽管这并不能保证只有一对作为父级):

  @EndpointsAliasProperty(setter=ParentSet, required=True)
  def parent(self):
    if self._parent is None and self.key is not None:
      self._parent = self.key.parent().string_id()
    return self._parent

完成此操作后,您无需更改任何 API 代码,示例将按预期工作。

于 2013-04-17T23:51:41.793 回答