anki/qt/aqt/tagedit.py

141 lines
4.7 KiB
Python
Raw Normal View History

2019-02-05 04:59:03 +01:00
# Copyright: Ankitects Pty Ltd and contributors
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
from __future__ import annotations
2012-12-22 01:17:10 +01:00
import re
from typing import Iterable
from anki.collection import Collection
from aqt import gui_hooks
2019-12-20 10:19:03 +01:00
from aqt.qt import *
class TagEdit(QLineEdit):
_completer: QCompleter | TagCompleter
lostFocus = pyqtSignal()
# 0 = tags, 1 = decks
def __init__(self, parent: QWidget, type: int = 0) -> None:
QLineEdit.__init__(self, parent)
self.col: Collection | None = None
self.model = QStringListModel()
self.type = type
if type == 0:
self._completer = TagCompleter(self.model, parent, self)
else:
self._completer = QCompleter(self.model, parent)
self._completer.setCompletionMode(QCompleter.PopupCompletion)
self._completer.setCaseSensitivity(Qt.CaseInsensitive)
self._completer.setFilterMode(Qt.MatchContains)
self.setCompleter(self._completer)
def setCol(self, col: Collection) -> None:
"Set the current col, updating list of available tags."
self.col = col
l: Iterable[str]
if self.type == 0:
l = self.col.tags.all()
else:
l = (d.name for d in self.col.decks.all_names_and_ids())
self.model.setStringList(l)
def focusInEvent(self, evt: QFocusEvent) -> None:
QLineEdit.focusInEvent(self, evt)
def keyPressEvent(self, evt: QKeyEvent) -> None:
if evt.key() in (Qt.Key_Up, Qt.Key_Down):
# show completer on arrow key up/down
if not self._completer.popup().isVisible():
self.showCompleter()
return
if evt.key() == Qt.Key_Tab and int(evt.modifiers()) & Qt.ControlModifier:
# select next completion
if not self._completer.popup().isVisible():
self.showCompleter()
index = self._completer.currentIndex()
self._completer.popup().setCurrentIndex(index)
cur_row = index.row()
if not self._completer.setCurrentRow(cur_row + 1):
self._completer.setCurrentRow(0)
return
if (
evt.key() in (Qt.Key_Enter, Qt.Key_Return)
and self._completer.popup().isVisible()
):
# apply first completion if no suggestion selected
selected_row = self._completer.popup().currentIndex().row()
if selected_row == -1:
self._completer.setCurrentRow(0)
index = self._completer.currentIndex()
self._completer.popup().setCurrentIndex(index)
self.hideCompleter()
QWidget.keyPressEvent(self, evt)
return
QLineEdit.keyPressEvent(self, evt)
if not evt.text():
# if it's a modifier, don't show
return
if evt.key() not in (
2019-12-23 01:34:10 +01:00
Qt.Key_Enter,
Qt.Key_Return,
Qt.Key_Escape,
Qt.Key_Space,
Qt.Key_Tab,
Qt.Key_Backspace,
Qt.Key_Delete,
):
self.showCompleter()
2020-04-01 09:13:08 +02:00
gui_hooks.tag_editor_did_process_key(self, evt)
def showCompleter(self) -> None:
self._completer.setCompletionPrefix(self.text())
self._completer.complete()
2021-02-02 14:30:53 +01:00
def focusOutEvent(self, evt: QFocusEvent) -> None:
QLineEdit.focusOutEvent(self, evt)
self.lostFocus.emit() # type: ignore
self._completer.popup().hide()
def hideCompleter(self) -> None:
if sip.isdeleted(self._completer):
return
self._completer.popup().hide()
2019-12-23 01:34:10 +01:00
class TagCompleter(QCompleter):
def __init__(
self,
model: QStringListModel,
parent: QWidget,
edit: TagEdit,
) -> None:
QCompleter.__init__(self, model, parent)
self.tags: list[str] = []
self.edit = edit
self.cursor: int | None = None
def splitPath(self, tags: str) -> list[str]:
stripped_tags = tags.strip()
stripped_tags = re.sub(" +", " ", stripped_tags)
self.tags = self.edit.col.tags.split(stripped_tags)
self.tags.append("")
p = self.edit.cursorPosition()
if tags.endswith(" "):
self.cursor = len(self.tags) - 1
else:
self.cursor = stripped_tags.count(" ", 0, p)
return [self.tags[self.cursor]]
def pathFromIndex(self, idx: QModelIndex) -> str:
if self.cursor is None:
return self.edit.text()
ret = QCompleter.pathFromIndex(self, idx)
self.tags[self.cursor] = ret
try:
self.tags.remove("")
except ValueError:
pass
return f"{' '.join(self.tags)} "