Updating files.
This commit is contained in:
parent
b0a3901df3
commit
7ce5a2bb8b
89
test.py
89
test.py
@ -1,89 +0,0 @@
|
|||||||
from anytree import Node
|
|
||||||
from anytree import RenderTree
|
|
||||||
from anytree import LevelOrderIter, LevelOrderGroupIter
|
|
||||||
from anytree.search import find, findall
|
|
||||||
|
|
||||||
cut = Node("cut")
|
|
||||||
cool = Node("cool", parent=cut)
|
|
||||||
press = Node("press into pan", parent=cool)
|
|
||||||
stir = Node("stir until coated", parent=press)
|
|
||||||
cereal = Node("crispy rice cereal (6 cups)", parent=stir)
|
|
||||||
melt_stir = Node("stir until melted", parent=stir)
|
|
||||||
marshmellows = Node("marshmellows (10 oz.)", parent=melt_stir)
|
|
||||||
melt = Node("melt", parent=melt_stir)
|
|
||||||
butter = Node("butter (3 tbsp)", parent=melt)
|
|
||||||
|
|
||||||
output_lines = {}
|
|
||||||
leaves = [node for node in sorted(cut.leaves, key=lambda node: node.depth, reverse=True)]
|
|
||||||
|
|
||||||
max_leaf_name_length = max([len(leaf.name) for leaf in leaves])
|
|
||||||
|
|
||||||
for leaf in leaves:
|
|
||||||
line = f"| {leaf.name.ljust(max_leaf_name_length)} |"
|
|
||||||
output_lines[leaf] = line
|
|
||||||
|
|
||||||
max_depth = max([leaf.depth for leaf in leaves])
|
|
||||||
|
|
||||||
for cur_depth in range(max_depth-1, -1, -1):
|
|
||||||
max_name_length = max([len(node.name) for node in findall(cut, lambda node: node.depth == cur_depth and not node in cut.leaves)])
|
|
||||||
max_name_length = max(max_name_length, 10)
|
|
||||||
for leaf in leaves:
|
|
||||||
task = find(cut, lambda node: node.depth == cur_depth and node in leaf.path)
|
|
||||||
if task and not task in cut.leaves:
|
|
||||||
output_lines[leaf] += f" {task.name.center(max_name_length)} |"
|
|
||||||
else:
|
|
||||||
output_lines[leaf] += f" {"".ljust(max_name_length)} "
|
|
||||||
|
|
||||||
for key in output_lines.keys():
|
|
||||||
print(output_lines[key])
|
|
||||||
|
|
||||||
print("")
|
|
||||||
|
|
||||||
output_lines = [""] * len(cut.leaves)
|
|
||||||
|
|
||||||
cur_line = 0
|
|
||||||
last_depth = cut.depth
|
|
||||||
for node in LevelOrderIter(cut):
|
|
||||||
if node in cut.leaves:
|
|
||||||
continue
|
|
||||||
if last_depth != node.depth:
|
|
||||||
cur_line = 0
|
|
||||||
|
|
||||||
output_lines[cur_line] = f" {node.name} " + output_lines[cur_line]
|
|
||||||
cur_line += len(node.children) + 1
|
|
||||||
last_depth = node.depth
|
|
||||||
|
|
||||||
for line in output_lines:
|
|
||||||
print(line)
|
|
||||||
|
|
||||||
rows = []
|
|
||||||
leaves = []
|
|
||||||
|
|
||||||
for i in range(len(cut.leaves)):
|
|
||||||
rows.append([])
|
|
||||||
leaves.append({"name":"wrong"})
|
|
||||||
|
|
||||||
cur_row = 0
|
|
||||||
for level in LevelOrderGroupIter(cut):
|
|
||||||
cur_row = 0
|
|
||||||
|
|
||||||
for node in reversed(level):
|
|
||||||
leaf_count = len(node.leaves)
|
|
||||||
if node in cut.leaves:
|
|
||||||
leaves[cur_row] = node
|
|
||||||
else:
|
|
||||||
rows[cur_row].append(f'<td rowspan="{leaf_count}">{node.name}</td>')
|
|
||||||
|
|
||||||
cur_row += max(leaf_count, 1)
|
|
||||||
last_depth = node.depth
|
|
||||||
|
|
||||||
for i, leaf in enumerate(leaves):
|
|
||||||
rows[i].append(f"<td>{leaf.name}</td>")
|
|
||||||
rows[i].reverse()
|
|
||||||
|
|
||||||
print("<tr>")
|
|
||||||
for cell in rows[i]:
|
|
||||||
print(cell)
|
|
||||||
print("</tr>")
|
|
||||||
|
|
||||||
|
|
Loading…
x
Reference in New Issue
Block a user