json - 使用 python 3 专门化 JSON 对象编码

Question

由于Python3中dict.values()和keys()的更改，我遇到了一些麻烦。

我的旧代码是这样的:

import json
class ComplexEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, complex):
            return [obj.real, obj.imag]
        return json.JSONEncoder.default(self, obj)

a = { '1' : 2 + 1j, '2' : 4 + 2j }

print(json.dumps(a.values(), cls=ComplexEncoder))

这在 Python 3.3+ 上会引发异常:

TypeError: dict_values([(2+1j), (4+2j)]) is not JSON serializable

简单的解决方法是执行 list(a.values())，对我来说问题是我在代码中有很多类似的实例。有没有办法扩展 ComplexEncoder 以便迭代 查看？

Answer 1

您可以将可迭代编码为列表:

class IterEncoder(json.JSONEncoder):
    def default(self, obj):
        try:
            return list(obj)
        except TypeError:
            return super().default(obj)

class ComplexEncoder(IterEncoder):
    def default(self, obj):
        if isinstance(obj, complex):
            return [obj.real, obj.imag]

        return super().default(obj)