Update AutoNode and TabSearch

Author: ArnoChenFx

NodeGraphQt/widgets/tab_search.py

@@ -153,10 +153,15 @@ def __init__(self, node_dict=None):
         self.line_edit.returnPressed.connect(self._on_search_submitted)
         self.line_edit.textChanged.connect(self._on_text_changed)
         self.rebuild = False
+        self._block_submit = False
 
     def __repr__(self):
         return '<{} at {}>'.format(self.__class__.__name__, hex(id(self)))
 
+    def keyPressEvent(self, event):
+        super(TabSearchMenuWidget, self).keyPressEvent(event)
+        self.line_edit.keyPressEvent(event)
+
     def _on_text_changed(self, text):
         self._clear_actions()
 
@@ -187,26 +192,29 @@ def _close(self):
         self._set_menu_visible(False)
         self.setVisible(False)
         self.menuAction().setVisible(False)
+        self._block_submit = True
 
     def _show(self):
         self.line_edit.setText("")
         self.line_edit.setFocus()
         self._set_menu_visible(True)
+        self._block_submit = False
         self.exec_(QtGui.QCursor.pos())
 
     def _on_search_submitted(self):
-        action = self.sender()
-        if type(action) is not QtWidgets.QAction:
-            if len(self._searched_actions) > 0:
-                action = self._searched_actions[0]
-            else:
-                self._close()
-                return
+        if not self._block_submit:
+            action = self.sender()
+            if type(action) is not QtWidgets.QAction:
+                if len(self._searched_actions) > 0:
+                    action = self._searched_actions[0]
+                else:
+                    self._close()
+                    return
 
-        text = action.text()
-        node_type = self._node_dict.get(text)
-        if node_type:
-            self.search_submitted.emit(node_type)
+            text = action.text()
+            node_type = self._node_dict.get(text)
+            if node_type:
+                self.search_submitted.emit(node_type)
 
         self._close()
 
@@ -223,6 +231,7 @@ def build_menu_tree(self):
                 if depth in menu_tree.keys():
                     if menu_name not in menu_tree[depth].keys():
                         new_menu = QtWidgets.QMenu(menu_name)
+                        new_menu.keyPressEvent = self.keyPressEvent
                         new_menu.setStyleSheet(STYLE_QMENU)
                         menu_tree[depth][menu_path] = new_menu
                 else:

example_auto_nodes/node_base/auto_node.py

@@ -81,7 +81,7 @@ def autoCook(self, mode):
     @property
     def cookTime(self):
         """
-        Get the last cooked time of the node.
+        Returns the last cooked time of the node.
         """
 
         return self._cookTime
@@ -98,6 +98,14 @@ def cookTime(self, cook_time):
         self._cookTime = cook_time
         self._update_tool_tip()
 
+    @property
+    def has_error(self):
+        """
+        Returns whether the node has errors.
+        """
+
+        return self._error
+
     def update_stream(self, forceCook=False):
         """
         Update all down stream nodes.
@@ -352,17 +360,11 @@ def _setup_tool_tip(self):
 
     def error(self, message=None):
         """
-        Update the node tooltip.
+        Set error message to node tooltip.
 
         Args:
-            message(str): the describe of the error or None.
-
-        Returns:
-            if message is None, returns whether the node has error.
+            message(str): the describe of the error.
         """
-
-        if message is None:
-            return self._error
         self._show_error(message)
 
     # def __del__(self):

example_auto_nodes/node_base/subgraph_node.py

@@ -168,7 +168,7 @@ def run(self):
                 node.when_disabled()
             else:
                 node.cook()
-            if node.error():
+            if node.has_error:
                 self.error("/"+node.view.toolTip())
                 break
 

example_auto_nodes/node_base/utils.py

@@ -7,7 +7,7 @@ def _update_nodes(nodes):
             node.when_disabled()
         else:
             node.cook()
-        if node.error():
+        if node.has_error:
             break