Python 未调用自定义JSON编码器

这可能应该被认为是json包的实现细节，也可能是Python版本的东西.在任何情况下，它对您的实施至关重要:

• dump()函数在内部调用编码器上的iterencode()方法(参见第169和176行in the actual source code).
• 然而，dumps()函数在内部调用encode()(参见第231和238行).

您可以通过调整encode()并覆盖JSONSerializer中的iterencode()来验证这一点，如下所示:

class JSONSerializer(json.JSONEncoder):
    def encode(self, obj):
        print("encode called")
        ...  # Your previous code here
        return super().encode(obj)

    def iterencode(self, *args, **kwargs):
        print("iterencode called")
        return super().iterencode(*args, **kwargs)

...你会看到只有"iterencode called"会与你的测试代码一起打印，而不是"encode called".

你在问题中链接的另一个堆栈溢出问题似乎也有同样的问题，至少在使用一个相当新版本的Python时(我目前正在编写这篇文章)——参见我的comment相应答案.

我有两个解决方案:

1. Either在你的Agent.memorize()方法中使用dumps()，例如:

   def memorize(self):
       with open("memory.json", "w") as w:
           w.write(json.dumps(self.G, cls=JSONSerializer))
   

2. Or将您自己的实现从encode()移动到iterencode()，例如:

   class JSONSerializer(json.JSONEncoder):
   
       def iterencode(self, obj, *args, **kwargs):
           if isinstance(obj, MutableMapping):
               for key in list(obj.keys()):
                   if isinstance(key, tuple):
                       strkey = "%d:%d" % key
                       obj[strkey] = obj.pop(key)
           yield from super().iterencode(obj, *args, **kwargs)
   

   这个第二种解决方案似乎有一个好处，即它可以与dump()和dumps()一起工作(见下面的注释).

一个补充说明:后来的dumps()函数似乎也会导致调用iterencode()(我没有跟踪源代码，以了解这到底发生在哪里，但从我添加的打印机来看，它肯定会发生).这有以下效果:(1)在第一个建议的解决方案中，由于首先调用encode()，我们可以进行所有的调整以使我们的数据可在那里串行化，在稍后的时间点，调用iterencode()将不会导致错误.(2)在第二个建议的解决方案中，当我们重新实现iterencode()时，我们的数据将在这一点上被设置为JSON可串行化.

Update:实际上还有第三种解决方案，感谢@ AbdulAzizBarkat的 comments :我们可以覆盖default()方法.然而，我们必须确保我们交出一个对象类型来进行序列化，而不是常规编码器处理，否则我们将永远无法使用它到达default()方法.在给定的代码中，我们可以将Agent实例本身传递给dump()或dumps()，而不是它的字典字段G.所以我们要做两个调整:

1. 将memorize()调整为self，而不是self.G，例如:

   def memorize(self):
        with open("memory.json", "w") as w:
            json.dump(self, w, cls=JSONSerializer)
   

2. 调整JSONSerializer.default()以处理Agent实例，例如，如下所示(我们不再需要encode()和iterencode()):

   class JSONSerializer(json.JSONEncoder):
       def default(self, obj):
           if isinstance(obj, Agent):
               new_obj = {}
               for key in obj.G.keys():
                   new_key = ("%d:%d" % key) if isinstance(key, tuple) else key
                   new_obj[new_key] = obj.G[key]
               return new_obj  # Return the adjusted dictionary
           return super().default(obj)
   

最后一点:在前两个解决方案中，原始词典的键被更改了.你可能不希望有这个，因为实际上你的实例不应该仅仅因为你转储了它的副本而改变.所以你可能更希望用改变后的键和原始值创建一个新的字典.我在第三个解决方案中解决了这个问题.