darkfi/bin/darkwallet/gui/print_tree.py

from pydrk import SceneNodeType, PropertyType
from .api import api

def join(parent_path, child_name):
    if parent_path == "/":
        return f"/{child_name}"
    return f"{parent_path}/{child_name}"

def print_tree():
    root_id = api.lookup_node_id("/")
    print("/")
    print_node_info(root_id, indent=1)

def print_node_info(parent_id, indent):
    ws = " "*4*indent
    for (child_name, child_id, child_type) in api.get_children(parent_id):
        match child_type:
            case SceneNodeType.ROOT:
                child_type = "root"
            case SceneNodeType.WINDOW:
                child_type = "window"
            case SceneNodeType.WINDOW_INPUT:
                child_type = "window_input"
            case SceneNodeType.KEYBOARD:
                child_type = "keyboard"
            case SceneNodeType.MOUSE:
                child_type = "mouse"
            case SceneNodeType.RENDER_LAYER:
                child_type = "render_layer"
            case SceneNodeType.RENDER_OBJECT:
                child_type = "render_object"
            case SceneNodeType.RENDER_MESH:
                child_type = "render_mesh"
            case SceneNodeType.RENDER_TEXT:
                child_type = "render_text"
            case SceneNodeType.RENDER_TEXTURE:
                child_type = "render_texture"
            case SceneNodeType.FONTS:
                child_type = "fonts"
            case SceneNodeType.FONT:
                child_type = "font"
            case SceneNodeType.LINE_POSITION:
                child_type = "line_position"

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

        print_node_info(child_id, indent+1)

    for prop in api.get_properties(parent_id):
        if prop.type != PropertyType.BUFFER:
            prop_val = api.get_property_value(parent_id, prop.name)

            if prop.type == PropertyType.STR:
                prop_val = [f"\"{pv}\"" for pv in prop_val]

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

            prop_val = f" = {prop_val}"
        else:
            prop_val = ""

        prop_type = PropertyType.to_str(prop.type)

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

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

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

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

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