add dynamic change the combobox widget menu

Author: ArnoChenFx

NodeGraphQt/base/node.py

@@ -704,6 +704,21 @@ def add_output(self, name='output', multi_output=True, display_name=True,
         self.model.outputs[port.name()] = port.model
         return port
 
+    def update_combo_menu(self, name, items):
+        if not self.has_property(name):
+            return
+        old_value = self.get_property(name)
+        self.set_property(name, items)
+        _name = '_'+name+"_"
+        if not self.has_property(_name):
+            self.create_property(_name, items)
+        else:
+            self.set_property(_name, items)
+        if old_value in items:
+            self.set_property(name, old_value)
+        else:
+            self.set_property(name, items[0])
+
     def inputs(self):
         """
         Returns all the input ports from the node.

NodeGraphQt/qgraphics/node_base.py

@@ -691,6 +691,9 @@ def get_widget(self, name):
             return widget
         raise NodeWidgetError('node has no widget "{}"'.format(name))
 
+    def has_widget(self, name):
+        return name in self._widgets.keys()
+
     def delete(self):
         for port, text in self._input_items.items():
             port.delete()

NodeGraphQt/widgets/node_widgets.py

@@ -192,6 +192,10 @@ def value(self):
 
     @value.setter
     def value(self, text=''):
+        if type(text) is list:
+            self._combo.clear()
+            self._combo.addItems(text)
+            return
         if text != self.value:
             index = self._combo.findText(text, QtCore.Qt.MatchExactly)
             self._combo.setCurrentIndex(index)

NodeGraphQt/widgets/properties.py

@@ -245,6 +245,9 @@ def get_value(self):
         return self.currentText()
 
     def set_value(self, value):
+        if type(value) is list:
+            self.set_items(value)
+            return
         if value != self.get_value():
             idx = self.findText(value, QtCore.Qt.MatchExactly)
             self.setCurrentIndex(idx)
@@ -871,7 +874,11 @@ def _read_node(self, node):
                 widget.setMinimumHeight(min_widget_height)
                 if prop_name in common_props.keys():
                     if 'items' in common_props[prop_name].keys():
-                        widget.set_items(common_props[prop_name]['items'])
+                        _prop_name = '_' + prop_name + "_"
+                        if node.has_property(_prop_name):
+                            widget.set_items(node.get_property(_prop_name))
+                        else:
+                            widget.set_items(common_props[prop_name]['items'])
                     if 'range' in common_props[prop_name].keys():
                         prop_range = common_props[prop_name]['range']
                         widget.set_min(prop_range[0])

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 Qt import QtCore
+from NodeGraphQt import QtCore
 import hashlib
 import copy
 import time
@@ -83,10 +83,12 @@ def getData(self, port):
         return self.get_property(port.name())
 
     def getInputData(self, port):
-        # get input data by input Port,the type of "port" can be :
-        # int : Port index
-        # str : Port name
-        # Port : Port object
+        """
+        get input data by input Port,the type of "port" can be :
+        int : Port index
+        str : Port name
+        Port : Port object
+        """
 
         if type(port) is int:
             to_port = self.input(port)
@@ -126,10 +128,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