__init__.py inside a Node library folder, recursively and automatically load all nodes.

Author: cesioarg

example_math_nodes.py

@@ -1,6 +1,6 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
-import example_nodes as Nodes
+from example_nodes import Nodes
 from NodeGraphQt import (NodeGraph,
                          BaseNode,
                          setup_context_menu)
@@ -40,12 +40,7 @@ def show_nodes_list(node):
     graph.node_double_clicked.connect(show_nodes_list)
 
     # registered nodes.
-    reg_nodes = [
-        Nodes.__dict__[n] for n in Nodes.__dict__
-        if hasattr(Nodes.__dict__[n], 'NODE_NAME')
-    ]
-
-    for n in reg_nodes:
+    for n in Nodes:
         graph.register_node(n)
 
     mathNodeA = graph.create_node('Math.MathFunctionsNode',

example_nodes/__init__.py

@@ -1,34 +1,28 @@
 #!/usr/bin/python
-# import os
-# import sys
-#
-# this_path = os.path.dirname(os.path.abspath(__file__))
-# directory = os.listdir(this_path)
-# modules = {}
-# for pkg_file in directory:
-#     pkg, ext = os.path.splitext(pkg_file)
-#     if pkg == '__init__':
-#         continue
-#     if ext in ('.py', '.pyc'):
-#         modules[pkg] = ('from . import {}'.format(pkg))
-#     elif os.path.isdir(os.path.join(directory, pkg_file)):
-#         modules[pkg] = 'from . import {}'.format(pkg)
-#
-# for pkg, pkg_imp in modules.items():
-#     try:
-#         exec pkg_imp
-#     except Exception as e:
-#         import traceback
-#         sys.stderr.write('Can\'t import module: {}'.format(pkg))
-#         info = sys.exc_info()
-#         type = info[0]
-#         param = info[1]
-#         tb = info[2]
-#         traceback.print_exception(type, param, tb, file=sys.stderr)
+import os
+import sys
+import inspect
+import importlib
 
-# Manually loading nodes
-from .math_node import MathFunctionsNode
-from .input_nodes import (DataInputNode, TickTimeNode, TextFileInputNode)
-from .viewer_nodes import DataViewerNode
-from .basic_nodes import (FooNode, BarNode)
-from .widget_nodes import (DropdownMenuNode, TextInputNode, CheckboxNode)
+
+def getNodesRecursively(path=__file__):
+    Nodes = []
+    basedir, filename = os.path.split(path)
+    for root, dirs, files in os.walk(basedir, topdown=False):
+        if root not in sys.path:
+            sys.path.append(root)
+
+        for name in files:
+            if name.endswith('.py') and not name.startswith('_'):
+                module_name = name[:-3]
+                module = importlib.import_module(module_name)
+                for name, obj in inspect.getmembers(module):
+                    if inspect.isclass(obj) and not obj.__name__ == 'BaseNode':
+                        for clsObj in inspect.getmro(obj):
+                            if clsObj.__name__ == 'BaseNode':
+                                Nodes.append(obj)
+                                break
+    return Nodes
+
+
+Nodes = getNodesRecursively()