新增python trick中的复杂dict打印功能

Author: pi-pi

Diff for: README.md

@@ -51,6 +51,7 @@ Code segment are often used in deep learning algorithms(pytorch/numpy)
    收集开发中常用的python稍微高级点的trick,
 1. 如何将dict参数自动变成类内部属性？
 2. 如何将字典变成可以属性访问?
+3. 复杂dict如何打印的漂亮，更有层次感？
 
 
 ### 7 numpy_trick

Diff for: docs/detail.md

@@ -101,6 +101,8 @@ print(iou)
 通过self.__ dict __.update实现
 2. 如何将字典变成可以属性访问?
 有两种方式：可以直接用第三方库addict实现；也可以自己简化写，核心就是递归+__ getattr __方法
+3. 复杂dict如何打印的漂亮，更有层次感？
+借助yapf库实现的pep8自动格式化字符串即可，我们要做的仅仅需要把dict按照行排列切割好就行，后面的每行如何格式化采用pep8标准自动执行
 
 
 ## 7 numpy_trick

Diff for: library/c6_python_trick.py

@@ -2,6 +2,10 @@
 # deep eyes
 
 # 汇集 python 相关的trick写法
+try:
+    from yapf.yapflib.yapf_api import FormatCode
+except:
+    FormatCode = None
 
 
 # 1. 如何将dict参数自动变成类内部属性？
@@ -35,3 +39,60 @@ def _hook(self, item):
         # 在.a和['a']时候自动调用
         def __getattr__(self, item):
             return self[item]
+
+
+# 3. 复杂dict如何打印的漂亮，更有层次感？
+# from mmcv 简化版本
+def pretty_text(src_dict):
+
+    def _format_basic_types(k, v):
+        if isinstance(v, str):
+            v_str = f"'{v}'"
+        else:
+            v_str = str(v)
+
+        attr_str = f'{str(k)}={v_str}'
+
+        return attr_str
+
+    def _format_list(k, v):
+        if all(isinstance(_, dict) for _ in v):
+            v_str = '[\n'
+            v_str += '\n'.join(
+                f'dict({_format_dict(v_)}),'
+                for v_ in v).rstrip(',')
+            attr_str = f'{str(k)}={v_str}'
+            attr_str = attr_str + ']'
+        else:
+            attr_str = _format_basic_types(k, v)
+        return attr_str
+
+    def _format_dict(input_dict, outest_level=False):
+        r = ''
+        s = []
+
+        for idx, (k, v) in enumerate(input_dict.items()):
+            end = '' if outest_level else ','  # 需要符合pep8格式
+            if isinstance(v, dict):
+                v_str = '\n' + _format_dict(v)  # 递归调用
+                attr_str = f'{str(k)}=dict({v_str}'
+                attr_str = attr_str + ')' + end
+            elif isinstance(v, list):
+                attr_str = _format_list(k, v) + end
+            else:
+                attr_str = _format_basic_types(k, v) + end
+
+            s.append(attr_str)
+        r += '\n'.join(s)
+        return r
+    # outest_level 最外层
+    # 仅仅需要把dict按照行排列切割好就行，后面的每行如何格式化采用pep8标准自动执行
+    text = _format_dict(src_dict, outest_level=True)
+    # 借助yapf库实现的pep8格式化字符串功能实现
+    yapf_style = dict(
+        based_on_style='pep8',
+        blank_line_before_nested_class_or_def=True,
+        split_before_expression_after_opening_paren=True)
+    text, _ = FormatCode(text, style_config=yapf_style, verify=True)
+
+    return text

Diff for: tools/c6_python_trick.py

@@ -21,6 +21,21 @@ def demo_2():
     print(cd.c.f.a)
 
 
+def demo_3():
+    cd = dict(a=1, b=[1, 2], c=dict(d=1, e=[2, 3], f=dict(a=1), aa=1, bb=1, cc=2, dd=dict(a=[1, 2, 3, 4],
+                                                                                          ee=dict(d=1, e=[2, 3],
+                                                                                                  f=dict(a=1), aa=1,
+                                                                                                  bb=1, cc=2, dd=dict(
+                                                                                                  a=[1, 2, 3, 4]))),
+                                    kk=dict(a=[1, 2, 3, 4], ee=dict(d=1, e=[2, 3],
+                                                                    f=dict(a=1), aa=1,
+                                                                    bb=1, cc=2, dd=dict(
+                                            a=[1, 2, 3, 4])))))
+    cd_str = pretty_text(cd)
+    print(cd_str)
+
+
 if __name__ == '__main__':
     # demo_1()
-    demo_2()
+    # demo_2()
+    demo_3()