We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 58021e6 commit 9ca9deeCopy full SHA for 9ca9dee
assets/17-it-generator/tree/step6/tree.py
@@ -1,14 +1,15 @@
1
-def tree(cls, level=0):
2
- yield cls.__name__, level
3
- for sub_cls in cls.__subclasses__():
4
- yield from tree(sub_cls, level+1)
+def tree(cls, dis_level, level=0):
+ if level <= dis_level:
+ yield cls.__name__, level
+ for sub_cls in cls.__subclasses__():
5
+ yield from tree(sub_cls, dis_level, level=level+1)
6
7
-def display(cls):
8
- for cls_name, level in tree(cls):
+def display(cls, display_level=5):
9
+ for cls_name, level in tree(cls, display_level - 1):
10
indent = ' ' * 4 * level
11
print(f'{indent}{cls_name}')
12
13
14
if __name__ == '__main__':
- display(BaseException)
15
+ display(BaseException, 2)
0 commit comments