The Computer as Master Mind Export JSON.py

import sys

#~ pip install anytree
from anytree import AnyNode, Node, PreOrderIter
from anytree.importer import JsonImporter, DictImporter
from anytree.exporter import JsonExporter

def nodenamefunc(node):
    if len(node.ancestors) == 0:
        return node.name

    Name = ""
    for i, anc in enumerate(node.ancestors):
        if i == 0:
            Name += anc.name
        else:
            Name += f":{anc.AnswerIdx}/{anc.name}"
    
    return Name + f":{node.AnswerIdx}/{node.name}"


def edgeattrfunc(node, child):
    return f'label="{GetBWStringLinear(child.AnswerIdx)}"'


def Export(Tree, FileName):
    Exp = JsonExporter()
    with open(FileName, "w") as f:
        Exp.write(Tree, f)

################### Main ###################

#~ Some helpers for indexing
filename = "The Computer as Master Mind Helpers.py" 
with open(filename, "rb") as source_file:
    code = compile(source_file.read(), filename, "exec")
exec(code)

#~ Import it from file
Imp = JsonImporter(DictImporter(nodecls=Node))
with open("ClassicMastermindTree.json", "r") as f:
    Root = Imp.read(f)

#~ Delete unnecessary attributes
for Node in PreOrderIter(Root):
    delattr(Node, "Codes")
    delattr(Node, "Pattern")

print(Root)

Export(Root, "ClassicMastermindTreeVis.json")

print("Done.")