use fuzzy find for tab search

Author: ArnoChenFx

NodeGraphQt/widgets/tab_search.py

@@ -1,7 +1,8 @@
 #!/usr/bin/python
 from .. import QtCore, QtWidgets, QtGui
-
 from .stylesheet import STYLE_TABSEARCH, STYLE_TABSEARCH_LIST, STYLE_QMENU
+from collections import OrderedDict
+import re
 
 
 class TabSearchCompleter(QtWidgets.QCompleter):
@@ -112,6 +113,18 @@ def set_nodes(self, node_dict=None):
         self._completer.setModel(self._model)
 
 
+def fuzzyFinder(key, collection):
+    suggestions = []
+    pattern = '.*?'.join(key.lower())
+    regex = re.compile(pattern)
+    for item in collection:
+        match = regex.search(item.lower())
+        if match:
+            suggestions.append((len(match.group()), match.start(), item))
+
+    return [x for _, _, x in sorted(suggestions)]
+
+
 class TabSearchMenuWidget(QtWidgets.QLineEdit):
     search_submitted = QtCore.Signal(str)
 
@@ -132,7 +145,7 @@ def __init__(self, parent=None, node_dict=None):
         self.SearchMenu.addAction(searchWidget)
         self.SearchMenu.setStyleSheet(STYLE_QMENU)
 
-        self._actions = []
+        self._actions = {}
         self._menus = {}
         self._searched_actions = []
 
@@ -151,30 +164,32 @@ def _on_text_changed(self,text):
 
         self._set_menu_visible(False)
 
-        self._searched_actions = [action for action in self._actions\
-                                 if text.lower() in action.text().lower()]
+        action_names = fuzzyFinder(text, self._actions.keys())
 
+        self._searched_actions = [self._actions[name] for name in action_names]
         self.SearchMenu.addActions(self._searched_actions)
 
+        if self._searched_actions:
+            self.SearchMenu.setActiveAction(self._searched_actions[0])
+
     def _clear_actions(self):
         for action in self._searched_actions:
             self.SearchMenu.removeAction(action)
         self._searched_actions = []
 
-    def _set_menu_visible(self,visible):
-        for menu in self._menus.values():
-            menu.menuAction().setVisible(visible)
+    def _set_menu_visible(self, visible):
+        [menu.menuAction().setVisible(visible) for menu in self._menus.values()]
 
     def _close(self):
         self._set_menu_visible(False)
         self.SearchMenu.setVisible(False)
         self.SearchMenu.menuAction().setVisible(False)
 
     def _show(self):
-        self.SearchMenu.exec_(QtGui.QCursor.pos())
         self.setText("")
         self.setFocus()
         self._set_menu_visible(True)
+        self.SearchMenu.exec_(QtGui.QCursor.pos())
 
     def _on_search_submitted(self):
         action = self.sender()
@@ -192,33 +207,52 @@ def _on_search_submitted(self):
 
         self._close()
 
-    def _generate_items_from_node_dict(self):
-        node_names = sorted(self._node_dict.keys())
+    def build_menu_tree(self):
         node_types = sorted(self._node_dict.values())
+        node_names = sorted(self._node_dict.keys())
+        menu_tree = OrderedDict()
 
-        self._menus.clear()
-        self._actions.clear()
-        self._searched_actions.clear()
-
+        max_depth = 0
         for node_type in node_types:
-            menu_name = ".".join(node_type.split(".")[:-1])
-            if menu_name not in self._menus.keys():
-                new_menu = QtWidgets.QMenu(menu_name)
-                new_menu.setStyleSheet(STYLE_QMENU)
-                self._menus[menu_name] = new_menu
-                self.SearchMenu.addMenu(new_menu)
+            trees = node_type.split(".")
+            trees.pop(-1)
+            for depth, menu_name in enumerate(trees):
+                menu_path = ".".join(trees[:depth+1])
+                if depth in menu_tree.keys():
+                    if menu_name not in menu_tree[depth].keys():
+                        new_menu = QtWidgets.QMenu(menu_name)
+                        new_menu.setStyleSheet(STYLE_QMENU)
+                        menu_tree[depth][menu_path] = new_menu
+                else:
+                    new_menu = QtWidgets.QMenu(menu_name)
+                    new_menu.setStyleSheet(STYLE_QMENU)
+                    menu_tree[depth] = {menu_path: new_menu}
+                if depth > 0:
+                    new_menu.parentPath = ".".join(trees[:depth])
+
+                max_depth = max(max_depth,depth)
+
+        for i in range(max_depth+1):
+            menus = menu_tree[i]
+            for menu_path, menu in menus.items():
+                self._menus[menu_path] = menu
+                if i == 0:
+                    self.SearchMenu.addMenu(menu)
+                else:
+                    parentMenu = self._menus[menu.parentPath]
+                    parentMenu.addMenu(menu)
 
         for name in node_names:
             action = QtWidgets.QAction(name, self)
             action.setText(name)
             action.triggered.connect(self._on_search_submitted)
-            self._actions.append(action)
+            self._actions[name] = action
 
             menu_name = self._node_dict[name]
-            menu_name = ".".join(menu_name.split(".")[:-1])
+            menu_path = ".".join(menu_name.split(".")[:-1])
 
-            if menu_name in self._menus.keys():
-                self._menus[menu_name].addAction(action)
+            if menu_path in self._menus.keys():
+                self._menus[menu_path].addAction(action)
             else:
                 self.SearchMenu.addAction(action)
 
@@ -231,7 +265,7 @@ def set_nodes(self, node_dict=None):
                     continue
                 for node_id in node_types:
                     self._node_dict['{} ({})'.format(name, node_id)] = node_id
-            self._generate_items_from_node_dict()
+            self.build_menu_tree()
 
         self._show()