langfingaz/anki
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

basic match highlighting

Browse Source
This commit is contained in:
Damien Elmes 2021-01-28 18:58:51 +10:00
parent 8b08687b0c
commit b148095243
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
qt/aqt/sidebar.py
View File

@ -195,7 +195,7 @@ class SidebarTreeView(QTreeView):
self.browser = browser
self.mw = browser.mw
self.col = self.mw.col
self.searching = False
self.current_search = False
self.setContextMenuPolicy(Qt.CustomContextMenu)
self.customContextMenuRequested.connect(self.onContextMenu) # type: ignore
@ -238,7 +238,7 @@ class SidebarTreeView(QTreeView):
# from PyQt5.QtTest import QAbstractItemModelTester
# tester = QAbstractItemModelTester(model)
self.searching = False
self.current_search = None
self.setModel(model)
expand_where_necessary(model, self)
@ -257,13 +257,23 @@ class SidebarTreeView(QTreeView):
else:
filter_model = self.model()
self.searching = True
self.current_search = text
# Without collapsing first, can be very slow. Surely there's
# a better way than this?
self.collapseAll()
filter_model.setFilterFixedString(text)
self.expandAll()
def drawRow(self, painter: QPainter, options: QStyleOptionViewItem, idx: QModelIndex):
if not self.current_search:
return super().drawRow(painter, options, idx)
if self.current_search.lower() in self.model().mapToSource(idx).internalPointer().name.lower():
brush = QBrush(QColor("lightyellow"))
painter.save()
painter.fillRect(options.rect, brush)
painter.restore()
return super().drawRow(painter, options, idx)
def onClickCurrent(self) -> None:
idx = self.currentIndex()
if idx.isValid():
@ -283,12 +293,12 @@ class SidebarTreeView(QTreeView):
super().keyPressEvent(event)
def onExpansion(self, idx: QModelIndex) -> None:
if self.searching:
if self.current_search:
return
self._onExpansionChange(idx, True)
def onCollapse(self, idx: QModelIndex) -> None:
if self.searching:
if self.current_search:
return
self._onExpansionChange(idx, False)