Anki/ankiqt/ui/tagedit.py
2008-12-04 02:35:41 +09:00

59 lines
1.9 KiB
Python

# Copyright: Damien Elmes <anki@ichi2.net>
# License: GNU GPL, version 3 or later; http://www.gnu.org/copyleft/gpl.html
from PyQt4.QtGui import *
from PyQt4.QtCore import *
from anki.utils import parseTags, canonifyTags, joinTags
class TagEdit(QLineEdit):
def __init__(self, parent, *args):
QLineEdit.__init__(self, parent, *args)
self.model = QStringListModel()
self.completer = TagCompleter(self.model, parent, self)
self.completer.setCompletionMode(QCompleter.UnfilteredPopupCompletion)
self.completer.setCaseSensitivity(Qt.CaseInsensitive)
self.setCompleter(self.completer)
def setDeck(self, deck, tags="user"):
"Set the current deck, updating list of available tags."
self.deck = deck
if tags == "user":
tags = self.deck.allUserTags()
else:
tags = self.deck.allTags()
self.model.setStringList(
QStringList(tags))
def focusOutEvent(self, evt):
QLineEdit.focusOutEvent(self, evt)
self.emit(SIGNAL("lostFocus"))
class TagCompleter(QCompleter):
def __init__(self, model, parent, edit, *args):
QCompleter.__init__(self, model, parent)
self.tags = []
self.edit = edit
def splitPath(self, str):
str = unicode(str)
if str.strip().startswith(","):
self.cursor = 0
return QStringList("")
self.tags = parseTags(str)
self.tags.append(u"")
p = self.edit.cursorPosition()
self.cursor = str.count(",", 0, p)
return QStringList(self.tags[self.cursor])
def pathFromIndex(self, idx):
ret = QCompleter.pathFromIndex(self, idx)
if u"" not in self.tags:
self.tags.append(u"")
self.tags[self.cursor] = unicode(ret)
try:
self.tags.remove(u"")
except ValueError:
pass
return ", ".join(self.tags)