darkfi/bin/app/pydrk/print_tree.py

from .api import Api, PropertyType, NODE_TYPE_NAMES, CallArgType, Expr

def print_tree(api, node_path="/", depth=None):
    print(node_path)
    print_node_info(api, node_path, depth, indent=1)

def print_node_info(api, parent_path, depth, indent):
    if indent - 1 == depth:
        return

    ws = " "*4*indent
    for (child_name, child_id, child_type) in api.get_children(parent_path):
        child_type = NODE_TYPE_NAMES.get(child_type, child_type)

        desc = f"{ws}{child_name}:{child_id}/"
        desc += " "*(50 - len(desc))
        desc += f"[{child_type}]"
        print(desc)

        if parent_path == "/":
            child_path = "/" + child_name
        else:
            child_path = parent_path + "/" + child_name

        print_node_info(api, child_path, depth, indent+1)

    for prop in api.get_properties(parent_path):
        prop_val = api.get_property_value(parent_path, prop.name)

        def fmt_str(pv):
            return pv if isinstance(pv, Expr) else f"\"{pv}\""

        def fmt_f32(pv):
            return pv if isinstance(pv, Expr) else f"{pv:.2f}"

        if prop.type == PropertyType.STR:
            prop_val = "[" + ", ".join(fmt_str(pv) for pv in prop_val) + "]"
        elif prop.type == PropertyType.FLOAT32:
            prop_val = "[" + ", ".join(fmt_f32(pv) for pv in prop_val) + "]"

        if len(prop_val) == 1:
            prop_val = prop_val[0]

        prop_val = f" = {prop_val}"

        prop_type = PropertyType.to_str(prop.type)

        print(f"{ws}{prop.name}: {prop_type}{prop_val}")
        if prop.depends:
            print(f"{ws}    depends: {prop.depends}")

    for sig in api.get_signals(parent_path):
        print(f"{ws}~{sig}")
        for slot_name, slot_id in api.get_slots(parent_path, sig):
            print(f"{ws}- '{slot_name}' ({slot_id})")

    for method_name in api.get_methods(parent_path):
        args, results = api.get_method(parent_path, method_name)

        args = [f"{name}: " + CallArgType.to_str(typ) for (name, _, typ) in args]
        results = [f"{name}: " + CallArgType.to_str(typ) for (name, _, typ) in (results or [])]

        method_str = f"{method_name}(" + ", ".join(args) + ") -> (" + ", ".join(results) + ")"
        print(f"{ws}{method_str}")