"""Functions in this module will be avaliable to call in jinja templates"""
import yaml

def recurse_class_tree_text(tree, indent = 4):
    return yaml.dump(tree, indent = indent).replace(": {}", "")

def flatten_struct(struct):
    out = {}
    for s in struct:
        key = list(s.keys())[0]
        out[key] = s[key]
    return out

def _get_helpers():
    import reflect
    import os

    r = reflect.Reflect(os.getcwd())
    r.import_module("jinja_helpers")
    return {k: v[0] for k, v in r.get_functions("jinja_helpers").items()}

if __name__ == "__main__":
    import json
    with open("100301654_report.json", "r") as f:
        init_struct = json.load(f)["files"]

    print(flatten_struct(flatten_struct(init_struct)["example.py"]["functions"]))