Question

我试图遍历深层嵌套/混合列表中的每个值，并将任何Decimal实例转换为字符串，以便我可以将它们存储在mongo中。

我的递归尝试达到了最大深度。我想以迭代或高效的方式解决这个问题。

这似乎不起作用，但这是我最近的尝试：

def convert_decimals(root_obj):
    objs_to_convert = [root_obj]

    while objs_to_convert:
        obj = objs_to_convert.pop(0)
        for k, v in enumerate(obj):

            if len(v):
                objs_to_convert.append(v)
            elif isinstance(v, Decimal):
                obj[k] = str(v)

示例输入：

[
    {
        'Payments': {
            'Payment': Decimal('495.64'),
            'IsCapped': True,
            'OtherFees': Decimal('0'),
            'CapCostTotal': Decimal('27900'),
            'Name': 'TestData',
            'Program': {
                'ProgramName': u'AST',
                'Description': None
            },
            'Rate': Decimal('0.0254'),
            'APR': Decimal('2.54'),
            'AppliedIds': [

            ],
            'Tax': Decimal('0')
        }
    }
]

转换后，十进制实例应为字符串

Answer 1

使用Decimal参数在json.dump()期间转换default：

import json
import sys
from decimal import Decimal

def default(obj):
    if isinstance(obj, Decimal):
        return str(obj)
    else:
        raise TypeError(obj)

# convert during dump
json.dump(data, sys.stdout, indent=2, default=default)

修改data inplace：

import json
import sys
from collections import MutableMapping, MutableSequence
from decimal import Decimal

def convert_decimal(json_data):
    stack = [json_data]
    while stack:
        json_data = stack.pop()
        if isinstance(json_data, MutableMapping): # json object
            it = json_data.items()
        elif isinstance(json_data, MutableSequence): # json array
            it = enumerate(json_data)
        else: # scalar data
            continue

        for k, v in it:
            if isinstance(v, Decimal):
                json_data[k] = str(v)
            else:
                stack.append(v)


# convert inplace
convert_decimal(data)
json.dump(data, sys.stdout, indent=2)

两个脚本都产生相同的输出：

[
  {
    "Payments": {
      "OtherFees": "0", 
      "APR": "2.54", 
      "Rate": "0.0254", 
      "IsCapped": true, 
      "Name": "TestData", 
      "Program": {
        "Description": null, 
        "ProgramName": "AST"
      }, 
      "AppliedIds": [], 
      "CapCostTotal": "27900", 
      "Tax": "0", 
      "Payment": "495.64"
    }
  }
]

如果您使用generic get_items() function：

，则可以简化convert_decimal()

def convert_decimal(json_data):
    stack = [json_data]
    while stack:
        json_data = stack.pop()
        for k, v in get_items(json_data):
            if isinstance(v, Decimal):
                json_data[k] = str(v)
            else:
                stack.append(v)

Answer 2

您想将小数转换为字符串，但递归地将您的函数应用于列表的内容和字典的值，否则返回对象不变？然后这样做：

def strip_decimals(o):
    if type(o) == Decimal:
        return str(o)
    elif type(o) == list:
        return map(strip_decimals, o)
    elif type(o) == dict:
        return dict([(k, strip_decimals(v)) for k, v in o.iteritems()])
    else:
        return o

结果：

[{'Payments': {'APR': '2.54',
           'AppliedIds': [],
           'CapCostTotal': '27900',
           'IsCapped': True,
           'Name': 'TestData',
           'OtherFees': '0',
           'Payment': '495.64',
           'Program': {'Description': None, 'ProgramName': u'AST'},
           'Rate': '0.0254',
           'Tax': '0'}}]

在深层嵌套和不可预测的列表中将python decimal转换为string

2 个答案: