创建自定义JSONEncoder

我正在运行Python 2.7，我试图创建一个JSONEncoder的自定义FloatEncoder子类。我遵循了很多例子，如this，但似乎没有任何工作。这里是我FloatEncoder类：创建自定义JSONEncoder

class FloatEncoder(JSONEncoder): 
    def _iterencode(self, obj, markers=None): 
     if isinstance(obj, float): 
      return (str(obj) for obj in [obj]) 
     return super(FloatEncoder, self)._iterencode(obj, markers)

而且这里是我打电话json.dumps：

with patch("utils.fileio.FloatEncoder") as float_patch: 
     for val,res in ((.0,'0.0012'),(.00009,'0.0001'),(0.99999,'1.0000'),({'hello':1.00001,'world':[True,1.00009]},'{"world": [true, 1.0001], "hello": 1.0000}')): 
      untrusted = dumps(val, cls=FloatEncoder) 
      self.assertTrue(float_patch._iterencode.called) 
      self.assertEqual(untrusted, res)

第一个断言失败，这意味着没有被执行_iterencode。在阅读JSON文档之后，我试着覆盖默认的（）方法，但那个也没有被调用。

来源

2011-06-20 Matt

FWIW，默认'（）'不会被调用，因为如果输入的是编码器默认支持的类型之一，它甚至不会看你的自定义的方法。在'_iterencode（）'的定义中比较'lib/json/encoder.py'：'_default（）'只在所有已知类型被覆盖之后在'else：'分支中被调用。因此，您无法覆盖已知类型的处理。 – Tomalak

您似乎在生成JSON（基于测试示例）时将浮点值舍入到小数点后4位。

JSONEncoder与Python 2.7一起发货没有_iterencode方法，所以这就是为什么它没有被调用。另外快速浏览一下json/encoder.py表明，这个类的编写方式很难改变float编码行为。或许，最好将问题分开，并在执行JSON序列化之前围绕浮点数。

编辑：亚历克斯·马尔泰利还提供在a related answer.猴子补丁解决方案与方法的问题是，你推出一个全球性的修改json库行为可能会不知不觉地影响一些其他的代码在你的应用程序是在假设浮点数没有四舍五入的情况下编写的。

试试这个：

from collections import Mapping, Sequence 
from unittest import TestCase, main 
from json import dumps 

def round_floats(o): 
    if isinstance(o, float): 
     return round(o, 4) 
    elif isinstance(o, basestring): 
     return o 
    elif isinstance(o, Sequence): 
     return [round_floats(item) for item in o] 
    elif isinstance(o, Mapping): 
     return dict((key, round_floats(value)) for key, value in o.iteritems()) 
    else: 
     return o 

class TestFoo(TestCase): 
    def test_it(self): 
     for val, res in ((.0, '0.0012'), 
         (.00009, '0.0001'), 
         (0.99999, '1.0'), 
         ({'hello': 1.00001, 'world': [True, 1.00009]}, 
          '{"world": [true, 1.0001], "hello": 1.0}')): 
      untrusted = dumps(round_floats(val)) 
      self.assertEqual(untrusted, res) 

if __name__ == '__main__': 
    main()

来源

2011-06-20 17:56:02

-1

不定义_iterencode，定义default，如该页上的第三个答案所示。

来源

2011-06-20 15:52:34

我也尝试过，但它是同样的问题：未调用default（）方法。 – Matt

创建自定义JSONEncoder

回答

相关问题