如何子类化 google app engine ndb 属性以支持 python 子类化对象

Question

来自这篇文章https://stackoverflow.com/a/32107024/5258689

我有一个 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() ？或者要扩展什么属性以便我可以创建它？

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

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的子类，返回一个@987654324 @instance 访问时。

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() 以便正确酸洗，否则会出现酸洗错误。