anki/qt/aqt/dyndeckconf.py

173 lines
5.4 KiB
Python
Raw Normal View History

2019-02-05 04:59:03 +01:00
# Copyright: Ankitects Pty Ltd and contributors
# -*- coding: utf-8 -*-
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
from typing import List, Optional
import aqt
from anki.lang import without_unicode_isolation
from anki.rsbackend import InvalidInput
2019-12-20 10:19:03 +01:00
from aqt.qt import *
from aqt.utils import (
TR,
askUser,
disable_help_button,
openHelp,
restoreGeom,
saveGeom,
show_invalid_search_error,
showWarning,
tr,
)
2019-12-20 10:19:03 +01:00
class DeckConf(QDialog):
def __init__(self, mw, first=False, search="", deck=None):
QDialog.__init__(self, mw)
self.mw = mw
self.deck = deck or self.mw.col.decks.current()
self.search = search
self.form = aqt.forms.dyndconf.Ui_Dialog()
self.form.setupUi(self)
if first:
label = tr(TR.DECKS_BUILD)
else:
label = tr(TR.ACTIONS_REBUILD)
2019-12-23 01:34:10 +01:00
self.ok = self.form.buttonBox.addButton(label, QDialogButtonBox.AcceptRole)
self.mw.checkpoint(tr(TR.ACTIONS_OPTIONS))
disable_help_button(self)
self.setWindowModality(Qt.WindowModal)
2020-10-09 18:38:07 +02:00
qconnect(self.form.buttonBox.helpRequested, lambda: openHelp("filtered-decks"))
self.setWindowTitle(
without_unicode_isolation(tr(TR.ACTIONS_OPTIONS_FOR, val=self.deck["name"]))
)
restoreGeom(self, "dyndeckconf")
2018-01-20 06:26:11 +01:00
self.initialSetup()
self.loadConf()
if search:
self.form.search.setText(search + " is:due")
self.form.search_2.setText(search + " is:new")
self.form.search.selectAll()
2018-01-14 10:20:01 +01:00
if self.mw.col.schedVer() == 1:
self.form.secondFilter.setVisible(False)
self.show()
self.exec_()
saveGeom(self, "dyndeckconf")
2018-01-20 06:26:11 +01:00
def initialSetup(self):
import anki.consts as cs
2019-12-23 01:34:10 +01:00
2020-11-17 10:23:06 +01:00
self.form.order.addItems(list(cs.dynOrderLabels(self.mw.col).values()))
self.form.order_2.addItems(list(cs.dynOrderLabels(self.mw.col).values()))
qconnect(self.form.resched.stateChanged, self._onReschedToggled)
2018-01-20 06:26:11 +01:00
def _onReschedToggled(self, _state):
2019-12-23 01:34:10 +01:00
self.form.previewDelayWidget.setVisible(
not self.form.resched.isChecked() and self.mw.col.schedVer() > 1
)
2018-01-20 06:26:11 +01:00
def loadConf(self):
f = self.form
d = self.deck
2019-12-23 01:34:10 +01:00
f.resched.setChecked(d["resched"])
2018-01-20 06:26:11 +01:00
self._onReschedToggled(0)
2019-12-23 01:34:10 +01:00
search, limit, order = d["terms"][0]
f.search.setText(search)
if self.mw.col.schedVer() == 1:
2019-12-23 01:34:10 +01:00
if d["delays"]:
f.steps.setText(self.listToUser(d["delays"]))
f.stepsOn.setChecked(True)
else:
f.steps.setVisible(False)
f.stepsOn.setVisible(False)
f.order.setCurrentIndex(order)
f.limit.setValue(limit)
2018-01-20 06:26:11 +01:00
f.previewDelay.setValue(d.get("previewDelay", 10))
2019-12-23 01:34:10 +01:00
if len(d["terms"]) > 1:
search, limit, order = d["terms"][1]
f.search_2.setText(search)
f.order_2.setCurrentIndex(order)
f.limit_2.setValue(limit)
f.secondFilter.setChecked(True)
f.filter2group.setVisible(True)
else:
f.order_2.setCurrentIndex(5)
f.limit_2.setValue(20)
f.secondFilter.setChecked(False)
f.filter2group.setVisible(False)
def saveConf(self):
f = self.form
d = self.deck
2019-12-23 01:34:10 +01:00
d["resched"] = f.resched.isChecked()
d["delays"] = None
if self.mw.col.schedVer() == 1 and f.stepsOn.isChecked():
steps = self.userToList(f.steps)
if steps:
2019-12-23 01:34:10 +01:00
d["delays"] = steps
else:
2019-12-23 01:34:10 +01:00
d["delays"] = None
search = self.mw.col.backend.normalize_search(f.search.text())
terms = [[search, f.limit.value(), f.order.currentIndex()]]
if f.secondFilter.isChecked():
search_2 = self.mw.col.backend.normalize_search(f.search_2.text())
terms.append([search_2, f.limit_2.value(), f.order_2.currentIndex()])
2019-12-23 01:34:10 +01:00
d["terms"] = terms
d["previewDelay"] = f.previewDelay.value()
self.mw.col.decks.save(d)
def reject(self):
self.ok = False
QDialog.reject(self)
def accept(self):
try:
self.saveConf()
except InvalidInput as err:
show_invalid_search_error(err)
return
if not self.mw.col.sched.rebuild_filtered_deck(self.deck["id"]):
2020-11-18 02:32:22 +01:00
if askUser(tr(TR.DECKS_THE_PROVIDED_SEARCH_DID_NOT_MATCH)):
return
self.mw.reset()
QDialog.accept(self)
# Step load/save - fixme: share with std options screen
########################################################
def listToUser(self, l):
return " ".join([str(x) for x in l])
def userToList(self, w, minSize=1) -> Optional[List[Union[float, int]]]:
items = str(w.text()).split(" ")
ret = []
for item in items:
if not item:
continue
try:
i = float(item)
assert i > 0
if i == int(i):
i = int(i)
ret.append(i)
except:
# invalid, don't update
showWarning(tr(TR.SCHEDULING_STEPS_MUST_BE_NUMBERS))
return None
if len(ret) < minSize:
showWarning(tr(TR.SCHEDULING_AT_LEAST_ONE_STEP_IS_REQUIRED))
return None
return ret