飘逸的python - 实现一个pretty函数美丽的输出嵌套字典

Posted 2020-09-16 cxchanpin

演示样例：

d = { "root": { "folder2": { "item2": None, "item1": None }, "folder1": { "subfolder1": { "item2": None, "item1": None }, "subfolder2": { "item3": None } } } }

美观输出是：

观察一下特点：

1.同一level的key左对齐,即indent同样
2.在,号后换行
3.假设value是字典。即嵌套字典，则嵌套字典在下一个level，每个level的key的indent都不一样

思路分析：
这是个"拼接字符串"问题,元素有""{}:,\n和空格indent.
遍历一个个(k, v)键值对,拼接yield起来,遇到嵌套字典就递归

即递归+yield

上代码。

#coding=utf-8
def pretty_dict(obj, indent=‘ ‘):
    def _pretty(obj, indent):
        for i, tup in enumerate(obj.items()):
            k, v = tup
            #假设是字符串则拼上""
            if isinstance(k, basestring): k = ‘"%s"‘% k
            if isinstance(v, basestring): v = ‘"%s"‘% v
            #假设是字典则递归
            if isinstance(v, dict):
                v = ‘‘.join(_pretty(v, indent + ‘ ‘* len(str(k) + ‘: {‘)))#计算下一层的indent
            #case,依据(k,v)对在哪个位置确定拼接什么
            if i == 0:#开头,拼左花括号
                if len(obj) == 1:
                    yield ‘{%s: %s}‘% (k, v)
                else:
                    yield ‘{%s: %s,\n‘% (k, v)
            elif i == len(obj) - 1:#结尾,拼右花括号
                yield ‘%s%s: %s}‘% (indent, k, v)
            else:#中间
                yield ‘%s%s: %s,\n‘% (indent, k, v)
    print ‘‘.join(_pretty(obj, indent))

d = { "root": { "folder2": { "item2": None, "item1": None }, "folder1": { "subfolder1": { "item2": None, "item1": None }, "subfolder2": { "item3": None } } } }
pretty_dict(d)