如何子类化 google App Engine ndb 属性 以支持 python 子类化对象

Question

来自这篇文章

我有一个 dict() 子类——它允许我做 dict.key（我的意思是使用点访问键）——如下：

class Permissions(dict):
"""
Example:
m = Map({'first_name': 'Eduardo'}, last_name='Pool', age=24, sports=['Soccer'])
"""
def __init__(self, *args, **kwargs):
    super(Permissions, self).__init__(*args, **kwargs)
    for arg in args:
        if isinstance(arg, dict):
            for k, v in arg.iteritems():
                self[k] = v

    if kwargs:
        for k, v in kwargs.iteritems():
            self[k] = v

def __getattr__(self, attr):
    return self.get(attr)

def __setattr__(self, key, value):
    self.__setitem__(key, value)

def __setitem__(self, key, value):
    super(Permissions, self).__setitem__(key, value)
    self.__dict__.update({key: value})

def __delattr__(self, item):
    self.__delitem__(item)

def __delitem__(self, key):
    super(Permissions, self).__delitem__(key)
    del self.__dict__[key]

我的问题是如何创建我自己的 PermessionsPropery()？或者 属性 要扩展什么以便我可以创建它？

我愿意在我的子类用户对象中使用这个 属性 添加学校名称作为键和权限作为字典值，例如（用户可以在多个学校拥有权限）：

from webapp2_extras.appengine.auth.models import User as webapp2User
class User(webapp2User):
    permissions = PermissionsProperty()

u = User(permissions=Permissions({"school1": {"teacher": True}}))

然后我检查用户的权限，例如：

if user.permissions[someshcool].teacher:
    #do stuff.....

#or
if user.permissions.someschool.teacher:
    #do stuff.....

我已尝试遵循此文档 https://cloud.google.com/appengine/docs/python/ndb/subclassprop 没有利润！

这有可能吗？如果是这样，怎么办？ 谢谢...

Answer 1

App Engine的ndb包不支持直接保存字典，但是json可以保存在JsonProperty中，字典很容易编码为json，所以最简单的实现是 JsonProperty 的子类，在访问时 returns 是一个 Permissions 实例。

class PermissionsProperty(ndb.JsonProperty):

    def _to_base_type(self, value):
        return dict(value)

    def _from_base_type(self, value):
        return Permissions(value)

虽然此实现不完整，因为 JsonProperty 将接受不是 Permissions 实例的值，因此您需要添加一个 _validate 方法以确保您保存的是正确类型的对象。

class PermissionsProperty(ndb.JsonProperty):

    def _to_base_type(self, value):
        return dict(value)

    def _from_base_type(self, value):
        return Permissions(value)

    def _validate(self, value):
        if not isinstance(value, Permissions):
            raise TypeError('Expected Permissions instance, got %r', % value)

Answer 2

正如@snakecharmerb 所说，它可以与 JasonProperty 一起使用，而且我在这里找到了另一种解决方案 ObjectProperty()

通过 pickle 支持任何 python 对象并将其保存到数据存储和反向！

下面是我的代码：

class Permissions(dict):
def __init__(self, *args, **kwargs):
    super(Permissions, self).__init__(*args, **kwargs)
    for arg in args:
        if isinstance(arg, dict):
            for k, v in arg.iteritems():
                self[k] = v

    if kwargs:
        for k, v in kwargs.iteritems():
            self[k] = v

def __getstate__(self):
    return self.__dict__

def __setstate__(self, d):
    self.__dict__.update(d)

def __getattr__(self, attr):
    return self.get(attr)

def __setattr__(self, key, value):
    self.__setitem__(key, value)

def __setitem__(self, key, value):
    super(Permissions, self).__setitem__(key, value)
    self.__dict__.update({key: value})

def __delattr__(self, item):
    self.__delitem__(item)

def __delitem__(self, key):
    super(Permissions, self).__delitem__(key)
    del self.__dict__[key]


class PermissionsProperty(ndb.PickleProperty):

    def _to_base_type(self, value):
        return pickle.dumps(value)

    def _from_base_type(self, value):
        return pickle.loads(value)

    def _validate(self, value):
        if not isinstance(value, Permissions):
            raise TypeError('Expected Permissions instance, got %r' % value)

请注意，我添加了 getstate() 和 setstate() 以便正确地酸洗，否则会出现酸洗错误。