python - 如何 JSON 序列化集合？

Question

我有一个 Python set，其中包含具有 __hash__ 和 __eq__ 方法的对象，以确保集合中不包含重复项。

我需要对这个结果 set 进行 json 编码，但是即使将空 set 传递给 json.dumps 方法也会引发 类型错误.

  File "/usr/lib/python2.7/json/encoder.py", line 201, in encode
    chunks = self.iterencode(o, _one_shot=True)
  File "/usr/lib/python2.7/json/encoder.py", line 264, in iterencode
    return _iterencode(o, 0)
  File "/usr/lib/python2.7/json/encoder.py", line 178, in default
    raise TypeError(repr(o) + " is not JSON serializable")
TypeError: set([]) is not JSON serializable

我知道我可以为具有自定义 default 方法的 json.JSONEncoder 类创建一个扩展，但我什至不确定从哪里开始转换设置。我应该使用默认方法中的 set 值创建一个字典，然后返回其编码吗？理想情况下，我想让默认方法能够处理原始编码器阻塞的所有数据类型(我使用 Mongo 作为数据源，因此日期似乎也会引发此错误)

任何正确方向的提示将不胜感激。

编辑:

感谢您的回答!也许我应该更准确。

我利用(并赞成)这里的答案来解决正在翻译的 set 的限制，但也有内部键是一个问题。

set 中的对象是转换为 __dict__ 的复杂对象，但它们本身也可以包含其属性的值，这些值可能不适用于json 编码器。

set 中有很多不同的类型，散列基本上计算实体的唯一 id，但在 NoSQL 的真正精神中，并不能确切说明子对象包含什么。

一个对象可能包含 starts 的日期值，而另一个对象可能有一些其他模式，其中不包含包含“非原始”对象的键。

这就是为什么我能想到的唯一解决方案是扩展 JSONEncoder 以替换 default 方法以打开不同的情况 - 但我不确定如何去解决这个问题，文档是模棱两可的。在嵌套对象中，从 default 返回的值是按键，还是只是查看整个对象的通用包含/丢弃？该方法如何适应嵌套值？我查看了以前的问题，似乎找不到针对特定情况进行编码的最佳方法(不幸的是，这似乎是我在这里需要做的)。

Answer 1

您可以创建一个自定义编码器，它在遇到 set 时会返回一个 list。这是一个例子:

>>> import json
>>> class SetEncoder(json.JSONEncoder):
...    def default(self, obj):
...       if isinstance(obj, set):
...          return list(obj)
...       return json.JSONEncoder.default(self, obj)
... 
>>> json.dumps(set([1,2,3,4,5]), cls=SetEncoder)
'[1, 2, 3, 4, 5]'

您也可以通过这种方式检测其他类型。如果您需要保留该列表实际上是一个集合，则可以使用自定义编码。 return {'type':'set', 'list':list(obj)} 之类的东西可能会起作用。

为了说明嵌套类型，请考虑将其序列化:

>>> class Something(object):
...    pass
>>> json.dumps(set([1,2,3,4,5,Something()]), cls=SetEncoder)

这会引发以下错误:

TypeError: <__main__.Something object at 0x1691c50> is not JSON serializable

这表示编码器将获取返回的 list 结果并递归调用其子级的序列化程序。要为多种类型添加自定义序列化程序，您可以这样做:

>>> class SetEncoder(json.JSONEncoder):
...    def default(self, obj):
...       if isinstance(obj, set):
...          return list(obj)
...       if isinstance(obj, Something):
...          return 'CustomSomethingRepresentation'
...       return json.JSONEncoder.default(self, obj)
... 
>>> json.dumps(set([1,2,3,4,5,Something()]), cls=SetEncoder)
'[1, 2, 3, 4, 5, "CustomSomethingRepresentation"]'