如何制作simplejson可序列化类

时间:2011-01-27 21:13:26

标签: python serialization simplejson

我有一个像这样定义的类

class A:
    def __init__(self):
        self.item1 = None
    def __repr__(self):
        return str(self.__dict__)

当我这样做时:

>>> import simplejson
>>> myA = A()
>>> simplejson.dumps(myA)
TypeError: {'item1': None} is not JSON serializable

我找不到原因。

我是否需要向A添加任何特定方法以使simplejson序列化我的类对象?

2 个答案:

答案 0 :(得分:10)

您无法使用simplejson序列化任意对象。您需要将defaultobject_hook传递给dumpload。这是一个例子:

class SerializerRegistry(object):
    def __init__(self):
        self._classes = {}
    def add(self, cls):
        self._classes[cls.__module__, cls.__name__] = cls
        return cls
    def object_hook(self, dct):
        module, cls_name = dct.pop('__type__', (None, None))
        if cls_name is not None:
            return self._classes[module, cls_name].from_dict(dct)
        else:
            return dct
    def default(self, obj):
        dct = obj.to_dict()
        dct['__type__'] = [type(obj).__module__,
                           type(obj).__name__]
        return dct

registry = SerializerRegistry()

@registry.add
class A(object):
    def __init__(self, item1):
        self.item1 = item1
    def __repr__(self):
        return str(self.__dict__)
    def to_dict(self):
        return dict(item1=self.item1)
    @classmethod
    def from_dict(cls, dct):
        return cls(**dct)

s = json.dumps(A(1), default=registry.default)
a = json.loads(s, object_hook=registry.object_hook)

结果如下:

>>> s
'{"item1": 1, "__type__": ["__main__", "A"]}'
>>> a
{'item1': 1}

但你真正需要的是一个函数default,它从你希望序列化的对象创建字典,以及一个函数object_hook,它在给出一个对象(正确类型)时返回它字典是不够的字典。最好的方法是在可序列化类上创建方法,这些方法从对象创建一个dict并将其构造回来,并且还有一个映射来识别字典所属的类。

您还可以在类中添加标识符,以用作_classes的索引。这样,如果必须移动一个类,就不会有问题。

答案 1 :(得分:3)

根据json module docs(simplejson在Python 2.6中被采用为json),您需要扩展json.JSONEncoder类,重写其默认方法,将对象转换为可序列化的类型。似乎没有一种方法可以在您的对象上查找。