解决python的json模块不喜欢循环引用

时间:2015-01-08 00:05:26

标签: python json

除了使用外部库(比如可能jsonpickle,虽然我还没试过),有没有办法让python的json模块到dump一个字典(或者列表等,有循环引用(只是删除引用,即)?

我只想使用json更容易看到一些调试输出。

1 个答案:

答案 0 :(得分:3)

好吧,除了标准模块之外什么都没有,这里有一个利用repr来处理循环引用的解决方案。编辑:最新消息,请参阅all-purpose function for dumping any python thing in a mostly-readable manner (aka dump)

# MAGIC-NUMBER: max length is just some guess at a reasonable size, e.g. 80 cols by 100 lines
def dump(value, msg='DUMP', max_length=80 * 100, stdout=False, pick=None):
    """
    Write as verbose of a description of the value as possible to logging.DEBUG.

    See http://stackoverflow.com/q/27830888/116891

    :param value: The item of interest.
    :type value: object
    :param msg: Prefix for the logged item (default='DUMP')
    :type msg: basestring
    :param max_length: Longest allowed string length (set to None for unlimited)
    :type max_length: int
    :param stdout: If true, print instead of logging (default=False)
    :type stdout: bool
    :param pick: If specified, dump only values for these keys of the item
        (value must be a dict or allow __dict__ access).
        The name comes from http://underscorejs.org/#pick.
    :type pick: iterable of basestring
    :return: True if message dumped
    :rtype: bool
    """
    if not logging.getLogger().isEnabledFor(logging.DEBUG) and not stdout:
        return

    if pick:
        d = value if isinstance(value, dict) else value.__dict__
        filtered = {
            property_name: d[property_name]
            for property_name in pick
            if property_name in d
        }
        value = filtered

    kwargs = dict(indent=2, sort_keys=True)
    try:
        import json
        info = json.dumps(value, **kwargs)
    except:
        # JSON doesn't like circular references :/
        try:
            string_repr = repr(value)
            # Replace python primitives, single-quotes, unicode, etc
            string_repr = string_repr\
                .replace('None', 'null')\
                .replace('True', 'true')\
                .replace('False', 'false')\
                .replace("u'", "'")\
                .replace("'", '"')

            # Replace object and function repr's like <MyObject ...>
            string_repr = re.sub(r':(\s+)(<[^>]+>)', r':\1"\2"', string_repr)

            # Replace tuples with lists, very naively
            string_repr = string_repr.replace('(', '[').replace(')', ']')

            info = json.dumps(json.loads(string_repr), **kwargs)
        except:
            from pprint import pformat
            info = pformat(value, indent=2)

    def _out(formatted_string, *format_args):
        """Format the string and output it to the correct location."""
        if stdout:
            print(formatted_string % format_args)
        else:
            logging.debug(formatted_string, *format_args)

    if max_length is None or len(info) <= max_length:
        _out('%s: %s', msg, info)
        return True
    else:
        _out(
            'Did not dump "%s" due to length restriction. Increase max_length if desired.', msg
        )
    return False