157 lines
5.2 KiB
Python
157 lines
5.2 KiB
Python
# Copyright: Ankitects Pty Ltd and contributors
|
|
# -*- coding: utf-8 -*-
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
import aqt
|
|
from anki.lang import _
|
|
from aqt import gui_hooks
|
|
from aqt.qt import *
|
|
from aqt.utils import getOnlyText, openHelp, restoreGeom, saveGeom, shortcut, showInfo
|
|
|
|
|
|
class StudyDeck(QDialog):
|
|
def __init__(
|
|
self,
|
|
mw,
|
|
names=None,
|
|
accept=None,
|
|
title=None,
|
|
help="studydeck",
|
|
current=None,
|
|
cancel=True,
|
|
parent=None,
|
|
dyn=False,
|
|
buttons=None,
|
|
geomKey="default",
|
|
) -> None:
|
|
QDialog.__init__(self, parent or mw)
|
|
if buttons is None:
|
|
buttons = []
|
|
self.mw = mw
|
|
self.form = aqt.forms.studydeck.Ui_Dialog()
|
|
self.form.setupUi(self)
|
|
self.form.filter.installEventFilter(self)
|
|
self.cancel = cancel
|
|
gui_hooks.state_did_reset.append(self.onReset)
|
|
self.geomKey = "studyDeck-" + geomKey
|
|
restoreGeom(self, self.geomKey)
|
|
if not cancel:
|
|
self.form.buttonBox.removeButton(
|
|
self.form.buttonBox.button(QDialogButtonBox.Cancel)
|
|
)
|
|
if buttons:
|
|
for b in buttons:
|
|
self.form.buttonBox.addButton(b, QDialogButtonBox.ActionRole)
|
|
else:
|
|
b = QPushButton(_("Add"))
|
|
b.setShortcut(QKeySequence("Ctrl+N"))
|
|
b.setToolTip(shortcut(_("Add New Deck (Ctrl+N)")))
|
|
self.form.buttonBox.addButton(b, QDialogButtonBox.ActionRole)
|
|
qconnect(b.clicked, self.onAddDeck)
|
|
if title:
|
|
self.setWindowTitle(title)
|
|
if not names:
|
|
names = [
|
|
d.name
|
|
for d in self.mw.col.decks.all_names_and_ids(
|
|
include_filtered=dyn, skip_empty_default=True
|
|
)
|
|
]
|
|
self.nameFunc = None
|
|
self.origNames = names
|
|
else:
|
|
self.nameFunc = names
|
|
self.origNames = names()
|
|
self.name = None
|
|
self.ok = self.form.buttonBox.addButton(
|
|
accept or _("Study"), QDialogButtonBox.AcceptRole
|
|
)
|
|
self.setWindowModality(Qt.WindowModal)
|
|
qconnect(self.form.buttonBox.helpRequested, lambda: openHelp(help))
|
|
qconnect(self.form.filter.textEdited, self.redraw)
|
|
qconnect(self.form.list.itemDoubleClicked, self.accept)
|
|
self.show()
|
|
# redraw after show so position at center correct
|
|
self.redraw("", current)
|
|
self.exec_()
|
|
|
|
def eventFilter(self, obj: QObject, evt: QEvent) -> bool:
|
|
if evt.type() == QEvent.KeyPress:
|
|
if evt.key() == Qt.Key_Up:
|
|
c = self.form.list.count()
|
|
row = self.form.list.currentRow() - 1
|
|
if row < 0:
|
|
row = c - 1
|
|
self.form.list.setCurrentRow(row)
|
|
return True
|
|
elif evt.key() == Qt.Key_Down:
|
|
c = self.form.list.count()
|
|
row = self.form.list.currentRow() + 1
|
|
if row == c:
|
|
row = 0
|
|
self.form.list.setCurrentRow(row)
|
|
return True
|
|
return False
|
|
|
|
def redraw(self, filt, focus=None):
|
|
self.filt = filt
|
|
self.focus = focus
|
|
self.names = [n for n in self.origNames if self._matches(n, filt)]
|
|
l = self.form.list
|
|
l.clear()
|
|
l.addItems(self.names)
|
|
if focus in self.names:
|
|
idx = self.names.index(focus)
|
|
else:
|
|
idx = 0
|
|
l.setCurrentRow(idx)
|
|
l.scrollToItem(l.item(idx), QAbstractItemView.PositionAtCenter)
|
|
|
|
def _matches(self, name, filt):
|
|
name = name.lower()
|
|
filt = filt.lower()
|
|
if not filt:
|
|
return True
|
|
for word in filt.split(" "):
|
|
if word not in name:
|
|
return False
|
|
return True
|
|
|
|
def onReset(self):
|
|
# model updated?
|
|
if self.nameFunc:
|
|
self.origNames = self.nameFunc()
|
|
self.redraw(self.filt, self.focus)
|
|
|
|
def accept(self) -> None:
|
|
saveGeom(self, self.geomKey)
|
|
gui_hooks.state_did_reset.remove(self.onReset)
|
|
row = self.form.list.currentRow()
|
|
if row < 0:
|
|
showInfo(_("Please select something."))
|
|
return
|
|
self.name = self.names[self.form.list.currentRow()]
|
|
QDialog.accept(self)
|
|
|
|
def reject(self) -> None:
|
|
saveGeom(self, self.geomKey)
|
|
gui_hooks.state_did_reset.remove(self.onReset)
|
|
QDialog.reject(self)
|
|
|
|
def onAddDeck(self) -> None:
|
|
row = self.form.list.currentRow()
|
|
if row < 0:
|
|
default = self.form.filter.text()
|
|
else:
|
|
default = self.names[self.form.list.currentRow()]
|
|
n = getOnlyText(_("New deck name:"), default=default)
|
|
if n:
|
|
self.mw.col.decks.id(n)
|
|
self.name = n
|
|
# make sure we clean up reset hook when manually exiting
|
|
gui_hooks.state_did_reset.remove(self.onReset)
|
|
if self.mw.state == "deckBrowser":
|
|
self.mw.deckBrowser.refresh()
|
|
gui_hooks.sidebar_should_refresh_decks()
|
|
QDialog.accept(self)
|