solve NODE_PROP_BUTTON function link issue

Author: ArnoChenFx

NodeGraphQt/base/model.py

@@ -92,7 +92,8 @@ def __init__(self):
         }
 
     def add_property(self, name, value, items=None, range=None,
-                     widget_type=NODE_PROP, tab='Properties', ext=None):
+                     widget_type=NODE_PROP, tab='Properties',
+                     ext=None, funcs=None):
         """
         add custom property.
 
@@ -104,6 +105,7 @@ def add_property(self, name, value, items=None, range=None,
             widget_type (int): widget type flag.
             tab (str): widget tab name.
             ext (str) file ext for NODE_PROP_FILE
+            funcs (list) functions for NODE_PROP_BUTTON
         """
         tab = tab or 'Properties'
 
@@ -125,6 +127,8 @@ def add_property(self, name, value, items=None, range=None,
                 self._TEMP_property_attrs[name]['range'] = range
             if ext:
                 self._TEMP_property_attrs[name]['ext'] = ext
+            if funcs:
+                self._TEMP_property_attrs[name]['funcs'] = funcs
         else:
             attrs = {self.type_: {name: {
                 'widget_type': widget_type,
@@ -136,6 +140,8 @@ def add_property(self, name, value, items=None, range=None,
                 attrs[self.type_][name]['range'] = range
             if ext:
                 attrs[self.type_][name]['ext'] = ext
+            if funcs:
+                attrs[self.type_][name]['funcs'] = funcs
             self._graph_model.set_node_common_properties(attrs)
 
     def set_property(self, name, value):
@@ -249,6 +255,7 @@ def to_dict(self):
                         json.dumps(v)
                     except:
                         to_remove.append(k)
+
             [custom_props.pop(k) for k in to_remove]
 
             node_dict['custom'] = custom_props

NodeGraphQt/base/node.py

@@ -237,7 +237,8 @@ def set_selected(self, selected=True):
         self.set_property('selected', selected)
 
     def create_property(self, name, value, items=None, range=None,
-                        widget_type=NODE_PROP, tab=None, ext=None):
+                        widget_type=NODE_PROP, tab=None, ext=None,
+                        funcs=None):
         """
         Creates a custom property to the node.
 
@@ -269,8 +270,9 @@ def create_property(self, name, value, items=None, range=None,
             widget_type (int): widget flag to display in the ``PropertiesBinWidget``
             tab (str): name of the widget tab to display in the properties bin.
             ext (str): file ext of ``NODE_PROP_FILE``
+            funcs (list) list of functions for NODE_PROP_BUTTON
         """
-        self.model.add_property(name, value, items, range, widget_type, tab, ext)
+        self.model.add_property(name, value, items, range, widget_type, tab, ext, funcs)
 
     def properties(self):
         """
@@ -567,11 +569,8 @@ def add_file_input(self, name, label='', text='', tab=None, ext="*"):
             tab (str): name of the widget tab to display in.
             ext (str): file ext
         """
-        self.model.add_property(name, text, None, None, NODE_PROP_FILE, tab, ext)
-        # use create_property will cause strange error
-        # self.create_property(
-        #     name, text, widget_type=NODE_PROP_FILE, tab=tab, ext=None)
-
+        self.create_property(
+            name, text, widget_type=NODE_PROP_FILE, tab=tab, ext=ext)
         widget = NodeFilePath(self.view, name, label, text, ext)
         widget.value_changed.connect(lambda k, v: self.set_property(k, v))
         self.view.add_widget(widget)

NodeGraphQt/widgets/properties.py

@@ -676,10 +676,12 @@ class PropButton(QtWidgets.QPushButton):
     def __init__(self, parent=None):
         super(PropButton, self).__init__(parent)
 
-    def set_value(self, value):
+    def set_value(self, value, node=None):
         # value: list of functions
+        if type(value) is not list:
+            return
         for func in value:
-            self.clicked.connect(func)
+            self.clicked.connect(lambda: func(node))
 
     def get_value(self):
         return None
@@ -876,6 +878,8 @@ def _read_node(self, node):
                         widget.set_max(prop_range[1])
                     if 'ext' in common_props[prop_name].keys():
                         widget.set_ext(common_props[prop_name]['ext'])
+                    if 'funcs' in common_props[prop_name].keys():
+                        widget.set_value(common_props[prop_name]['funcs'], node)
 
                 prop_window.add_widget(prop_name, widget, value,
                                        prop_name.replace('_', ' '))

example_auto_nodes/node_base/auto_node.py

@@ -1,7 +1,7 @@
 from NodeGraphQt.base.node import BaseNode
 from NodeGraphQt.base.port import Port
 from NodeGraphQt.constants import NODE_PROP
-from NodeGraphQt import QtCore
+from Qt import QtCore
 import hashlib
 import copy
 import time
@@ -25,7 +25,7 @@ def get(self, text, Min=50, Max=200):
         return self.colors[text]
 
 
-class AutoNode(BaseNode,QtCore.QObject):
+class AutoNode(BaseNode, QtCore.QObject):
     cooked = QtCore.Signal()
 
     def __init__(self, defaultInputType=None, defaultOutputType=None):
@@ -75,7 +75,8 @@ def cookTime(self, time):
     def cookNextNode(self):
         for nodeList in self.connected_output_nodes().values():
             for n in nodeList:
-                n.cook()
+                if n is not self:
+                    n.cook()
 
     def getData(self, port):
         # for custom output data
@@ -125,10 +126,10 @@ def cook(self, forceCook=False):
 
         _start_time = time.time()
 
-        try:
-            self.run()
-        except Exception as error:
-            self.error(error)
+        #try:
+        self.run()
+        #except Exception as error:
+        #    self.error(error)
 
         self._autoCook = _tmp
 
@@ -202,8 +203,8 @@ def set_port_type(self, port, value_type):
             current_port.view.setToolTip('{}: {} ({}) '.format(current_port.name(), data_type_name, conn_type))
 
     def create_property(self, name, value, items=None, range=None,
-                        widget_type=NODE_PROP, tab=None):
-        super(AutoNode, self).create_property(name, value, items, range, widget_type, tab)
+                        widget_type=NODE_PROP, tab=None, ext=None, funcs=None):
+        super(AutoNode, self).create_property(name, value, items, range, widget_type, tab, ext, funcs)
 
         if value is not None:
             self.set_port_type(name, type(value))
@@ -235,7 +236,6 @@ def set_disabled(self, mode=False):
         else:
             self.cook()
 
-
     def _close_error(self):
         self._error = False
         self.set_property('color', self.defaultColor)
@@ -250,7 +250,7 @@ def _show_error(self, message):
         tooltip = '<font color="red"><br>({})</br></font>'.format(message)
         self._update_tool_tip(tooltip)
 
-    def _update_tool_tip(self, message = None):
+    def _update_tool_tip(self, message=None):
         if message is None:
             tooltip = self._toolTip.format(self._cookTime)
         else:
@@ -269,3 +269,8 @@ def error(self, message=None):
             return self._error
 
         self._show_error(message)
+
+    def update_model(self):
+        if self.error():
+            self.set_property('color', self.defaultColor)
+        super(AutoNode, self).update_model()