36 lines
935 B
Python
Executable File
36 lines
935 B
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from tree import tree
|
|
|
|
|
|
SP = '\N{SPACE}'
|
|
HLIN = '\N{BOX DRAWINGS LIGHT HORIZONTAL}' # ─
|
|
ELBOW = f'\N{BOX DRAWINGS LIGHT UP AND RIGHT}{HLIN*2}{SP}' # └──
|
|
TEE = f'\N{BOX DRAWINGS LIGHT VERTICAL AND RIGHT}{HLIN*2}{SP}' # ├──
|
|
PIPE = f'\N{BOX DRAWINGS LIGHT VERTICAL}{SP*3}' # │
|
|
|
|
|
|
def cls_name(cls):
|
|
module = 'builtins.' if cls.__module__ == 'builtins' else ''
|
|
return module + cls.__name__
|
|
|
|
def render_lines(tree_iter):
|
|
cls, _, _ = next(tree_iter)
|
|
yield cls_name(cls)
|
|
prefix = ''
|
|
|
|
for cls, level, last in tree_iter:
|
|
prefix = prefix[:4 * (level-1)]
|
|
prefix = prefix.replace(TEE, PIPE).replace(ELBOW, SP*4)
|
|
prefix += ELBOW if last else TEE
|
|
yield prefix + cls_name(cls)
|
|
|
|
|
|
def draw(cls):
|
|
for line in render_lines(tree(cls)):
|
|
print(line)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
draw(Exception)
|