if True: xxx = 'xxx' from PySide import QtGui def toolbarAction(act): print('show toolbar ' + act['toolbar'] + ' from workbenches ' + repr(act['workbenches'])) def toolAction(act): print('start action for tool ' + act['toolbar'] + '.' + act['tool'] + ' from workbenches ' + repr(act['workbenches'])) def subToolAction(act): print('start action for tool ' + act['toolbar'] + '.' + act['tool'] + '.' + act['subTool'] + ' from workbenches ' + repr(act['workbenches'])) def documentObjectAction(): print('select object ' + o.Document.Name + '.' + o.Name) def documentAction(): print('switch to document ' + o.Document.Name) actionHandlers = { 'toolbarAction': toolbarAction, 'toolAction': toolAction, 'subToolAction': subToolAction, 'documentObjectAction': documentObjectAction, 'documentAction': documentAction } # Inspired by https://stackoverflow.com/a/7767999/324969 class SearchQCompleter(QtGui.QCompleter): def __init__(self, model, itemDelegate): super(SearchQCompleter, self).__init__() super(SearchQCompleter, self).setModel(QtCore.QIdentityProxyModel()) super(SearchQCompleter, self).popup().setItemDelegate(itemDelegate) self.setModel(model) def setModel(self, itemGroups): self.itemGroups = itemGroups self.filterModel('') def filterModel(self, userInput): def matches(s): return userInput.lower() in s.lower() def filterGroup(group): if matches(group['text']): # If a group matches, include the entire subtree (might need to disable this if it causes too much noise) return group else: subitems = filterGroups(group['subitems']) if len(subitems) > 0 or matches(group['text']): return { 'text': group['text'], 'icon': group['icon'], 'action': group['action'], 'subitems': subitems } else: return None def filterGroups(groups): groups = (filterGroup(group) for group in groups) return [group for group in groups if group is not None] def addGroups(filteredGroups, depth=0): for group in filteredGroups: mdl.appendRow([QtGui.QStandardItem(group['icon'] or QtGui.QIcon(), group['text']), QtGui.QStandardItem(str(depth)), QtGui.QStandardItem(group['action'])]) addGroups(group['subitems'], depth+1) mdl = QtGui.QStandardItemModel() mdl.appendColumn([]) addGroups(filterGroups(itemGroups)) print('setSourceModel for userInput ' + repr(userInput) + ' with ' + str(mdl.rowCount()) + ' rows.') self.model().setSourceModel(mdl) # https://stackoverflow.com/a/65930408/324969 # the splitPath(self, path) method is called every time the input string changes, before # drawing the completion list, we latch onto this method to also update the model to contain # the appropriate results, as given by the custom filterAcceptsRow method above. def splitPath(self, path): self.filterModel(path) # Pretend that the user endered the empty string, so that all items from the filteredProxyModel match. return '' # # Inspired by https://forum.qt.io/topic/69807/qtreeview-indent-entire-row class IndentedItemDelegate(QtGui.QStyledItemDelegate): def __init__(self): super(IndentedItemDelegate, self).__init__() def paint(self, painter, option, index): depth = int(option.widget.model().itemData(index.siblingAtColumn(1))[0]) indent = 16 * depth option.rect.adjust(indent, 0, 0, 0) super(IndentedItemDelegate, self).paint(painter, option, index) # class SearchBox(QtGui.QLineEdit): resultSelected = QtCore.Signal(int, str) def __init__(self, itemGroups): super(SearchBox, self).__init__() qom = SearchQCompleter(itemGroups, IndentedItemDelegate()) qom.setMaxVisibleItems(20) #qom.setCompletionMode(QtGui.QCompleter.CompletionMode.PopupCompletion) # Thanks to https://saurabhg.com/programming/search-box-using-qlineedit/ for indicating a few useful options ico = QtGui.QIcon(':/icons/help-browser.svg') #ico = QtGui.QIcon(':/icons/WhatsThis.svg') self.addAction(ico, QtGui.QLineEdit.LeadingPosition) self.setClearButtonEnabled(True) self.setPlaceholderText('Search tools, prefs & tree') self.setFixedWidth(200) # Signals & slots for enter / click def completerActivated(index): print('fooooooooooooo') global xxx xxx = index print(qom.model().rowCount(), index.row()) # TODO: run the action! result = str(qom.model().data(index.siblingAtColumn(1))) print('res='+result) self.clear() self.completer().setCompletionPrefix('') self.resultSelected.emit(str(index), result) def returnPressed(): #self.clear() #self.completer().setCompletionPrefix('') pass #text = sea.text() #self.clear() #self.resultSelected.emit('text returnPressed' + text) self.returnPressed.connect(returnPressed) #QtCore.QObject.connect(self.completer(), QtCore.SIGNAL('activated(QModelIndex)'), completerActivated) #, QtCore.Qt.ConnectionType.QueuedConnection) qom.activated.connect(completerActivated, QtCore.Qt.ConnectionType.DirectConnection) #, QtCore.Qt.ConnectionType.QueuedConnection) #self.completer().activated.connect(returnPressedOrCompleterActivated) def textChanged(): print('textChanged') # Workaround: Clear completion prefix and still show the completion box when doing backspace after typing a single character if self.text() == '': self.completer().setCompletionPrefix(self.text()) self.completer().complete() self.textChanged.connect(textChanged) self.setCompleter(qom) def focusInEvent(self, e): super(SearchBox, self).focusInEvent(e) self.completer().setCompletionPrefix(self.text()) self.completer().complete() self.completer().setCurrentRow(1) # Does not work #d=QtGui.QKeyEvent(QtCore.QEvent.KeyPress, QtCore.Qt.Key_Down, QtCore.Qt.NoModifier) #QtCore.QCoreApplication.postEvent(self, d) def mousePressEvent(self, e): super(SearchBox, self).mousePressEvent(e) self.completer().setCompletionPrefix(self.text()) self.completer().complete() self.completer().setCurrentRow(1) # Does not work #d=QtGui.QKeyEvent(QtCore.QEvent.KeyPress, QtCore.Qt.Key_Down, QtCore.Qt.NoModifier) #QtCore.QCoreApplication.postEvent(self, d) sea = QtGui.QLineEdit() # mw = Gui.getMainWindow() mdi = mw.findChild(QtGui.QMdiArea) wdg = QtGui.QWidget() lay = QtGui.QGridLayout(wdg) mwx = QtGui.QMainWindow() mbr = mw.findChildren(QtGui.QToolBar, 'File')[0] all_tbs = set() for wbname, workbench in Gui.listWorkbenches().items(): try: tbs = workbench.listToolbars() except: continue # careful, tbs contains all the toolbars of the workbench, including shared toolbars for tb in tbs: all_tbs.add(tb) import json def serializeIcon(icon): iconPixmaps = {} for sz in icon.availableSizes(): strW = str(sz.width()) strH = str(sz.height()) iconPixmaps[strW] = {} iconPixmaps[strW][strH] = {} for strMode, mode in {'normal':QtGui.QIcon.Mode.Normal, 'disabled':QtGui.QIcon.Mode.Disabled, 'active':QtGui.QIcon.Mode.Active, 'selected':QtGui.QIcon.Mode.Selected}.items(): iconPixmaps[strW][strH][strMode] = {} for strState, state in {'off':QtGui.QIcon.State.Off, 'on':QtGui.QIcon.State.On}.items(): buf = QtCore.QBuffer() buf.open(QtCore.QIODevice.WriteOnly) icon.pixmap(sz, mode, state).save(buf, 'PNG') iconPixmaps[strW][strH][strMode][strState] = QtCore.QTextCodec.codecForName('UTF-8').toUnicode(buf.data().toBase64()) return iconPixmaps # workbenches is a list(str), toolbar is a str, text is a str, icon is a QtGui.QIcon def serializeTool(tool): return { 'workbenches': tool['workbenches'], 'toolbar': tool['toolbar'], 'text': tool['text'], 'icon': serializeIcon(tool['icon']) } def deserializeIcon(iconPixmaps): ico = QtGui.QIcon() for strW, wPixmaps in iconPixmaps.items(): for strH, hPixmaps in wPixmaps.items(): for strMode, modePixmaps in hPixmaps.items(): mode = {'normal':QtGui.QIcon.Mode.Normal, 'disabled':QtGui.QIcon.Mode.Disabled, 'active':QtGui.QIcon.Mode.Active, 'selected':QtGui.QIcon.Mode.Selected}[strMode] for strState, statePixmap in modePixmaps.items(): state = {'off':QtGui.QIcon.State.Off, 'on':QtGui.QIcon.State.On}[strState] pxm = QtGui.QPixmap() pxm.loadFromData(QtCore.QByteArray.fromBase64(QtCore.QTextCodec.codecForName('UTF-8').fromUnicode(statePixmap))) ico.addPixmap(pxm, mode, state) return ico def deserializeTool(tool): return { 'workbenches': tool['workbenches'], 'toolbar': tool['toolbar'], 'text': tool['text'], 'icon': deserializeIcon(tool['icon']) } #serialized = serializeTools([{'workbenches': ['wb1', 'wb2'], 'toolbar': 'tbr', 'text': 'aa', 'icon': ic}]) #serialized = serializeTools([]) #TODO:save in App.getUserAppDataDir() ################################################################################################################ def GatherTools(): itemGroups = [] for toolbarName in all_tbs: for the_toolbar in mw.findChildren(QtGui.QToolBar, toolbarName): group = [] for tbt in the_toolbar.findChildren(QtGui.QToolButton): text = tbt.text() act = tbt.defaultAction() if text != '': # TODO: there also is the tooltip icon = tbt.icon() #sim.appendRow([QtGui.QStandardItem(icon, 't:' + text), QtGui.QStandardItem('tool')]) men = tbt.menu() subgroup = [] if men: subgroup = [] for mac in men.actions(): if mac.text(): #all_actions.append(mac.trigger) action = { 'handler': 'subToolAction', 'workbenches': [], 'toolbar': toolbarName, 'tool': text, 'subtool': mac.text() } #print('whaaaat', str(len(all_actions))) subgroup.append({'icon':mac.icon(), 'text':mac.text(), 'action':json.dumps(action), 'subitems':[]}) #all_actions.append(tbt.actions().trigger) #global lalala #lalala=tbt #print('whuuuut', str(len(all_actions))) action = { 'handler': 'toolAction', 'workbenches': [], 'toolbar': toolbarName, 'tool': text } group.append({'icon':icon, 'text':text, 'action': json.dumps(action), 'subitems': subgroup}) #viu = mw.findChildren(QtGui.QToolBar, 'View')[0] #tbt = viu.findChildren(QtGui.QToolButton) #men = tbt[3].menu() #acs = men.actions() # QtGui.QAction list #act = tbt[2].defaultAction() # QtGui.QAction #act.trigger() action = { 'handler': 'toolbarAction', 'workbenches': [], 'toolbar': toolbarName } itemGroups.append({ 'icon': QtGui.QIcon(':/icons/Group.svg'), 'text': toolbarName, 'action': json.dumps(action), 'subitems': group }) # def document(doc): group = [] for o in doc.Objects: #all_actions.append(lambda: Gui.Selection.addSelection(o.Document.Name, o.Name)) action = { 'handler': 'documentObjectAction', 'document': o.Document.Name, 'object': o.Name } item = { 'icon': o.ViewObject.Icon if o.ViewObject and o.ViewObject.Icon else None, 'text': o.Label + ' (' + o.Name + ')', 'action': json.dumps(action), 'subitems': [] } group.append(item) # Todo: this should also select the document in the tree view action = { 'handler': 'documentAction', 'document': o.Document.Name } itemGroups.append({ 'icon': QtGui.QIcon(':/icons/Document.svg'), 'text': doc.Label + ' (' + doc.Name + ')', 'action':json.dumps(action), 'subitems': group }) if App.ActiveDocument: document(App.ActiveDocument) for docname, doc in App.listDocuments().items(): if not App.activeDocument or docname != App.ActiveDocument.Name: document(doc) return itemGroups def serializeItemGroup(itemGroup): return { 'icon': serializeIcon(itemGroup['icon']), 'text': itemGroup['text'], 'action': itemGroup['action'], 'subitems': serializeItemGroups(itemGroup['subitems']) } def serializeItemGroups(itemGroups): return [serializeItemGroup(itemGroup) for itemGroup in itemGroups] def serialize(itemGroups): return json.dumps(serializeItemGroups(itemGroups)) def deserializeItemGroup(itemGroup): #print('dIG' + text + " : " + repr(itemGroup['icon'])[:100] + "......................." + repr(itemGroup['icon'])[-100:]) return { 'icon': deserializeIcon(itemGroup['icon']), 'text': itemGroup['text'], 'action': itemGroup['action'], 'subitems': deserializeItemGroups(itemGroup['subitems']) } def deserializeItemGroups(serializedItemGroups): return [deserializeItemGroup(itemGroup) for itemGroup in serializedItemGroups] def deserialize(serializeItemGroups): return deserializeItemGroups(json.loads(serializedItemGroups)) serializedItemGroups = serialize(GatherTools()) itemGroups = deserialize(serializedItemGroups) # sea = SearchBox(itemGroups) sea.resultSelected.connect(lambda x, y: print('aaa' + repr(y) + 'end')) wax = QtGui.QWidgetAction(None) wax.setDefaultWidget(sea) #mbr.addWidget(sea) mbr.addAction(wax)