export_taxonomy.py

from anytree.exporter import DotExporter

COLOR_SCHEME = ["aliceblue", "antiquewhite", "azure", "coral", "palegreen"]
    
def export(self):
    """
    Export tree into graphviz format in both *.dot file and image *.png file
    """

    def nodeattr_fn(node):
        return f'style=filled color={COLOR_SCHEME[node.depth]}'

    def nodename_fn(node):
        name = node.name
        return name

    root = self.nodes[self.root_key]
    dot = DotExporter(
        root,
        nodenamefunc=nodename_fn,
        nodeattrfunc=nodeattr_fn,
        options=[
            'graph [layout = dot, ranksep="1.5", nodesep="0.7"]',
            'rankdir ="LR"',
            'node [fontname="Arial"]'
        ]
    )
    dot.to_dotfile(f"./{self.root_key}.dot")
    dot.to_picture(f"./{self.root_key}.png")