feat: receive custom depth on rich.pretty ✨

CONTRIBUTORS.md

@@ -4,6 +4,7 @@ The following people have contributed to the development of Rich:
 
 <!-- Add your name below, sort alphabetically by surname. Link to Github profile / your home page. -->
 
+- [Meysam Azad](https://github.com/meysam81)
 - [Gregory Beauregard](https://github.com/GBeauregard/pyffstream)
 - [Pete Davison](https://github.com/pd93)
 - [Oleksis Fraga](https://github.com/oleksis)

rich/pretty.py

@@ -1,3 +1,4 @@
+import argparse
 import builtins
 import os
 from rich.repr import RichReprResult
@@ -821,6 +822,10 @@ def __repr__(self) -> str:
     }
     data["foo"].append(data)  # type: ignore
 
+    parser = argparse.ArgumentParser()
+    parser.add_argument("-d", "--depth", type=int, default=4)
+    parsed = parser.parse_args(sys.argv[1:])
+
     from rich import print
 
-    print(Pretty(data, indent_guides=True, max_string=20))
+    print(Pretty(data, indent_guides=True, max_string=20, indent_size=parsed.depth))

tests/test_pretty.py

@@ -152,6 +152,40 @@ def test_indent_lines():
     assert result == expected
 
 
+def test_custom_depth():
+    custom_indent = 16
+
+    data = {
+        "key": "value",
+        "list": ["el1", "el2"],
+        "dict": {"list": ["sub-el1", "sub-el2"]},
+    }
+    expected = """{
+│               'key': 'value',
+│               'list': [
+│               │               'el1',
+│               │               'el2'
+│               ],
+│               'dict': {
+│               │               'list': [
+│               │               │               'sub-el1',
+│               │               │               'sub-el2'
+│               │               ]
+│               }
+}
+"""
+
+    console = Console(width=100, color_system=None)
+    console.begin_capture()
+    console.print(
+        Pretty(data, indent_guides=True, expand_all=True, indent_size=custom_indent)
+    )
+    result = console.end_capture()
+    print(repr(result))
+    print(result)
+    assert result == expected
+
+
 def test_pprint():
     console = Console(color_system=None)
     console.begin_capture()