Source code for citelang.main.graph.tree

__author__ = "Vanessa Sochat"
__copyright__ = "Copyright 2022, Vanessa Sochat"
__license__ = "MPL 2.0"

import random

from rich import print
from rich.text import Text
from rich.tree import Tree

# Randomly select an AMAZING icon
# Note some of these require TWO spaces after the icon to render properly!
# The faded heart is my favorite, so we want it 1/2 the time
icons = ["⭐️ ", "✨️ ", "💛️ ", "🔷️ ", "❤️  ", "💜️ ", "🧡️ ", "🔴️ ", "♦️  ", "🔶️ "] + [
    "💗️ "
] * 10
root_icons = ["🦄️", "☕️", "🤓️", "⭐️"]


[docs]def print_tree(root):

    # Randomly select icons
    icon = random.choice(icons)
    root_icon = random.choice(root_icons)

    node = Text(f"{root_icon} {root.name}", "bold magenta")
    credit = round(root.credit, root.round_by)
    node.append(f" ({credit})", "blue")
    tree = Tree(
        node,
        guide_style="bold bright_blue",
    )

    generate_tree(root, tree=tree, icon=icon)
    print(tree)


[docs]def generate_tree(next_node, icon, tree=None) -> None:
    """
    Recursively build a Tree with a dependency result
    """
    for child in next_node.children:
        # We won't have a tree on the first run
        credit = round(child.credit, child.round_by)
        node = Text(child.name, "green")
        node.highlight_regex(r"\..*$", "bold red")
        node.stylize(f"link file://{child.name}")
        node.append(f" ({credit})", "blue")
        branch = tree.add(Text(icon) + node)
        generate_tree(child, tree=branch, icon=icon)