Skip to content

Commit 17f1a53

Browse files
committed
register nodes func
1 parent 147e3e8 commit 17f1a53

File tree

3 files changed

+14
-5
lines changed

3 files changed

+14
-5
lines changed

NodeGraphQt/base/factory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def register_node(self, node, alias=None):
6767
register the node.
6868
6969
Args:
70-
node (Node): node item
70+
node (NodeGraphQt.NodeObject): node object.
7171
alias (str): custom alias for the node identifier (optional).
7272
"""
7373
if node is None:

NodeGraphQt/base/graph.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -828,7 +828,7 @@ def registered_nodes(self):
828828

829829
def register_node(self, node, alias=None):
830830
"""
831-
Register the node to the node graph vendor.
831+
Register the node to the :meth:`NodeGraph.node_factory
832832
833833
Args:
834834
node (NodeGraphQt.NodeObject): node.
@@ -837,6 +837,16 @@ def register_node(self, node, alias=None):
837837
self._node_factory.register_node(node, alias)
838838
self._viewer.rebuild_tab_search()
839839

840+
def register_nodes(self, nodes):
841+
"""
842+
Register the nodes to the :meth:`NodeGraph.node_factory
843+
844+
Args:
845+
nodes (list[NodeGraphQt.NodeObject]): list of nodes.
846+
"""
847+
[self._node_factory.register_node(n) for n in nodes]
848+
self._viewer.rebuild_tab_search()
849+
840850
def create_node(self, node_type, name=None, selected=True, color=None,
841851
text_color=None, pos=None):
842852
"""

example.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -168,16 +168,15 @@ def show_nodes_list(node):
168168

169169

170170
# registered nodes.
171-
reg_nodes = [
171+
nodes_to_reg = [
172172
BackdropNode, MyNode,
173173
basic_nodes.FooNode,
174174
basic_nodes.BarNode,
175175
widget_nodes.DropdownMenuNode,
176176
widget_nodes.TextInputNode,
177177
widget_nodes.CheckboxNode
178178
]
179-
for n in reg_nodes:
180-
graph.register_node(n)
179+
graph.register_nodes(nodes_to_reg)
181180

182181
my_node = graph.create_node('com.chantasticvfx.MyNode',
183182
name='chantastic!',

0 commit comments

Comments
 (0)