Commit f152417

committed

refactored drawtree.py

1 parent 980f750 commit f152417Copy full SHA for f152417

File tree

+31

-38

lines changed

+31

-38

lines changed

Lines changed: 28 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,28 @@`
	`1`	`+from tree import tree`
	`2`	`+`
	`3`	`+SP = '\N{SPACE}'`
	`4`	`+HLIN = '\N{BOX DRAWINGS LIGHT HORIZONTAL}' # ─`
	`5`	`+ELBOW = f'\N{BOX DRAWINGS LIGHT UP AND RIGHT}{HLIN*2}{SP}' # └──`
	`6`	`+TEE = f'\N{BOX DRAWINGS LIGHT VERTICAL AND RIGHT}{HLIN*2}{SP}' # ├──`
	`7`	`+PIPE = f'\N{BOX DRAWINGS LIGHT VERTICAL}{SP*3}' # │`
	`8`	`+`
	`9`	`+`
	`10`	`+def render_lines(tree_iter):`
	`11`	`+ name, _, _ = next(tree_iter)`
	`12`	`+ yield name`
	`13`	`+ prefix = ''`
	`14`	`+`
	`15`	`+ for name, level, last in tree_iter:`
	`16`	`+ prefix = prefix[:4 * (level-1)]`
	`17`	`+ prefix = prefix.replace(TEE, PIPE).replace(ELBOW, SP*4)`
	`18`	`+ prefix += ELBOW if last else TEE`
	`19`	`+ yield prefix + name`
	`20`	`+`
	`21`	`+`
	`22`	`+def draw(cls):`
	`23`	`+ for line in render_lines(tree(cls)):`
	`24`	`+ print(line)`
	`25`	`+`
	`26`	`+`
	`27`	`+if __name__ == '__main__':`
	`28`	`+ draw(BaseException)`

Lines changed: 0 additions & 35 deletions

This file was deleted.

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from pretty_tree import tree, render_lines`
	`1`	`+from drawtree import tree, render_lines`
`2`	`2`
`3`	`3`	`def test_1_level():`
`4`	`4`	`result = list(render_lines(tree(BrokenPipeError)))`
`@@ -59,7 +59,7 @@ class Leaf2(Branch): pass`
`59`	`59`	`assert expected == result`
`60`	`60`
`61`	`61`
`62`		`-def test_3_levels_2_leaves():`
	`62`	`+def test_3_levels_2_leaves_dedent():`
`63`	`63`	`class A: pass`
`64`	`64`	`class B(A): pass`
`65`	`65`	`class C(B): pass`
`@@ -77,7 +77,7 @@ class E(D): pass`
`77`	`77`	`assert expected == result`
`78`	`78`
`79`	`79`
`80`		`-def test_4_levels_4_leaves():`
	`80`	`+def test_4_levels_4_leaves_dedent():`
`81`	`81`	`class A: pass`
`82`	`82`	`class B1(A): pass`
`83`	`83`	`class C1(B1): pass`

Comments

(0)