__init__.py

import importlib
import os

NODE_CLASS_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS = {}

# Main nodes for all users
NODE_MODULES = [
  ".nodes"
]

def load_nodes(module_name: str):
    global NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS

    module = importlib.import_module(module_name, package=__name__)

    NODE_CLASS_MAPPINGS = {
        **NODE_CLASS_MAPPINGS,
        **module.NODE_CLASS_MAPPINGS,
    }
    NODE_DISPLAY_NAME_MAPPINGS = {
        **NODE_DISPLAY_NAME_MAPPINGS,
        **module.NODE_DISPLAY_NAME_MAPPINGS,
    }

def write_nodes_list(module_names: list[str]):
    this_dir = os.path.dirname(os.path.abspath(__file__))
    path = os.path.join(this_dir, "nodes.log")

    lines = []

    for module_name in module_names:
        module = importlib.import_module(module_name, package=__name__)

        lines.append(module_name.strip("."))

        for identifier, display_name in module.NODE_DISPLAY_NAME_MAPPINGS.items():
            lines.append(f"  {identifier}: {display_name}")

        lines.append("")

    lines = "\n".join(lines)

    with open(path, "w", encoding="utf8") as f:
        f.write(lines)


for module_name in NODE_MODULES:
    load_nodes(module_name)