anki/qt/aqt/customstudy.py

216 lines
8.0 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
import aqt
from anki.collection import NamedFilter
from anki.consts import *
2019-12-20 10:19:03 +01:00
from aqt.qt import *
from aqt.utils import TR, disable_help_button, showInfo, showWarning, tr
RADIO_NEW = 1
RADIO_REV = 2
RADIO_FORGOT = 3
RADIO_AHEAD = 4
RADIO_PREVIEW = 5
RADIO_CRAM = 6
TYPE_NEW = 0
TYPE_DUE = 1
TYPE_REVIEW = 2
TYPE_ALL = 3
2019-12-23 01:34:10 +01:00
class CustomStudy(QDialog):
def __init__(self, mw) -> None:
QDialog.__init__(self, mw)
self.mw = mw
self.deck = self.mw.col.decks.current()
self.conf = self.mw.col.decks.get_config(self.deck["conf"])
self.form = f = aqt.forms.customstudy.Ui_Dialog()
self.created_custom_study = False
f.setupUi(self)
disable_help_button(self)
self.setWindowModality(Qt.WindowModal)
self.setupSignals()
2020-01-31 01:53:52 +01:00
f.radioNew.click()
self.exec_()
def setupSignals(self):
f = self.form
qconnect(f.radioNew.clicked, lambda: self.onRadioChange(RADIO_NEW))
qconnect(f.radioRev.clicked, lambda: self.onRadioChange(RADIO_REV))
qconnect(f.radioForgot.clicked, lambda: self.onRadioChange(RADIO_FORGOT))
qconnect(f.radioAhead.clicked, lambda: self.onRadioChange(RADIO_AHEAD))
qconnect(f.radioPreview.clicked, lambda: self.onRadioChange(RADIO_PREVIEW))
qconnect(f.radioCram.clicked, lambda: self.onRadioChange(RADIO_CRAM))
def onRadioChange(self, idx):
2019-12-23 01:34:10 +01:00
f = self.form
sp = f.spin
smin = 1
smax = DYN_MAX_SIZE
sval = 1
post = tr(TR.CUSTOM_STUDY_CARDS)
tit = ""
spShow = True
typeShow = False
ok = tr(TR.CUSTOM_STUDY_OK)
2019-12-23 01:34:10 +01:00
def plus(num):
if num == 1000:
num = "1000+"
2019-12-23 01:34:10 +01:00
return "<b>" + str(num) + "</b>"
if idx == RADIO_NEW:
new = self.mw.col.sched.totalNewForCurrentDeck()
# get the number of new cards in deck that exceed the new cards limit
2019-12-23 01:34:10 +01:00
newUnderLearning = min(
new, self.conf["new"]["perDay"] - self.deck["newToday"][1]
)
newExceeding = min(new, new - newUnderLearning)
tit = tr(
TR.CUSTOM_STUDY_NEW_CARDS_IN_DECK_OVER_TODAY, val=plus(newExceeding)
)
pre = tr(TR.CUSTOM_STUDY_INCREASE_TODAYS_NEW_CARD_LIMIT_BY)
2019-12-23 01:34:10 +01:00
sval = min(new, self.deck.get("extendNew", 10))
smin = -DYN_MAX_SIZE
smax = newExceeding
elif idx == RADIO_REV:
rev = self.mw.col.sched.totalRevForCurrentDeck()
# get the number of review due in deck that exceed the review due limit
2019-12-23 01:34:10 +01:00
revUnderLearning = min(
rev, self.conf["rev"]["perDay"] - self.deck["revToday"][1]
)
revExceeding = min(rev, rev - revUnderLearning)
tit = tr(
TR.CUSTOM_STUDY_REVIEWS_DUE_IN_DECK_OVER_TODAY, val=plus(revExceeding)
)
pre = tr(TR.CUSTOM_STUDY_INCREASE_TODAYS_REVIEW_LIMIT_BY)
2019-12-23 01:34:10 +01:00
sval = min(rev, self.deck.get("extendRev", 10))
smin = -DYN_MAX_SIZE
smax = revExceeding
elif idx == RADIO_FORGOT:
pre = tr(TR.CUSTOM_STUDY_REVIEW_CARDS_FORGOTTEN_IN_LAST)
post = tr(TR.SCHEDULING_DAYS)
smax = 30
elif idx == RADIO_AHEAD:
pre = tr(TR.CUSTOM_STUDY_REVIEW_AHEAD_BY)
post = tr(TR.SCHEDULING_DAYS)
elif idx == RADIO_PREVIEW:
pre = tr(TR.CUSTOM_STUDY_PREVIEW_NEW_CARDS_ADDED_IN_THE)
post = tr(TR.SCHEDULING_DAYS)
sval = 1
elif idx == RADIO_CRAM:
pre = tr(TR.CUSTOM_STUDY_SELECT)
post = tr(TR.CUSTOM_STUDY_CARDS_FROM_THE_DECK)
2019-12-23 01:34:10 +01:00
# tit = _("After pressing OK, you can choose which tags to include.")
ok = tr(TR.CUSTOM_STUDY_CHOOSE_TAGS)
sval = 100
typeShow = True
sp.setVisible(spShow)
f.cardType.setVisible(typeShow)
f.title.setText(tit)
f.title.setVisible(not not tit)
f.spin.setMinimum(smin)
f.spin.setMaximum(smax)
if smax > 0:
f.spin.setEnabled(True)
else:
f.spin.setEnabled(False)
f.spin.setValue(sval)
f.preSpin.setText(pre)
f.postSpin.setText(post)
f.buttonBox.button(QDialogButtonBox.Ok).setText(ok)
self.radioIdx = idx
def accept(self):
2019-12-23 01:34:10 +01:00
f = self.form
i = self.radioIdx
spin = f.spin.value()
if i == RADIO_NEW:
2019-12-23 01:34:10 +01:00
self.deck["extendNew"] = spin
self.mw.col.decks.save(self.deck)
self.mw.col.sched.extendLimits(spin, 0)
self.mw.reset()
return QDialog.accept(self)
elif i == RADIO_REV:
2019-12-23 01:34:10 +01:00
self.deck["extendRev"] = spin
self.mw.col.decks.save(self.deck)
self.mw.col.sched.extendLimits(0, spin)
self.mw.reset()
return QDialog.accept(self)
elif i == RADIO_CRAM:
tags = self._getTags()
# the rest create a filtered deck
cur = self.mw.col.decks.byName(tr(TR.CUSTOM_STUDY_CUSTOM_STUDY_SESSION))
if cur:
2019-12-23 01:34:10 +01:00
if not cur["dyn"]:
showInfo(tr(TR.CUSTOM_STUDY_MUST_RENAME_DECK))
return QDialog.accept(self)
else:
# safe to empty
2020-09-03 09:43:07 +02:00
self.mw.col.sched.empty_filtered_deck(cur["id"])
# reuse; don't delete as it may have children
dyn = cur
2019-12-23 01:34:10 +01:00
self.mw.col.decks.select(cur["id"])
else:
did = self.mw.col.decks.new_filtered(
tr(TR.CUSTOM_STUDY_CUSTOM_STUDY_SESSION)
)
dyn = self.mw.col.decks.get(did)
# and then set various options
if i == RADIO_FORGOT:
search = self.mw.col.search_string(forgot_in=spin)
dyn["terms"][0] = [search, DYN_MAX_SIZE, DYN_RANDOM]
2019-12-23 01:34:10 +01:00
dyn["resched"] = False
elif i == RADIO_AHEAD:
search = self.mw.col.search_string(due_in=spin)
dyn["terms"][0] = [search, DYN_MAX_SIZE, DYN_DUE]
2019-12-23 01:34:10 +01:00
dyn["resched"] = True
elif i == RADIO_PREVIEW:
search = self.mw.col.search_string(name=NamedFilter.NEW, added_in=spin)
dyn["terms"][0] = [search, DYN_MAX_SIZE, DYN_OLDEST]
2019-12-23 01:34:10 +01:00
dyn["resched"] = False
elif i == RADIO_CRAM:
type = f.cardType.currentRow()
if type == TYPE_NEW:
terms = self.mw.col.search_string(name=NamedFilter.NEW)
ord = DYN_ADDED
2019-12-23 01:34:10 +01:00
dyn["resched"] = True
elif type == TYPE_DUE:
terms = self.mw.col.search_string(name=NamedFilter.DUE)
ord = DYN_DUE
2019-12-23 01:34:10 +01:00
dyn["resched"] = True
elif type == TYPE_REVIEW:
terms = self.mw.col.search_string(negate=True, name=NamedFilter.NEW)
ord = DYN_RANDOM
2019-12-23 01:34:10 +01:00
dyn["resched"] = True
else:
terms = ""
ord = DYN_RANDOM
2019-12-23 01:34:10 +01:00
dyn["resched"] = False
dyn["terms"][0] = [(terms + tags).strip(), spin, ord]
# add deck limit
dyn["terms"][0][0] = self.mw.col.search_string(
deck=self.deck["name"], searches=[dyn["terms"][0][0]]
)
2020-05-22 03:25:11 +02:00
self.mw.col.decks.save(dyn)
# generate cards
self.created_custom_study = True
if not self.mw.col.sched.rebuild_filtered_deck(dyn["id"]):
return showWarning(tr(TR.CUSTOM_STUDY_NO_CARDS_MATCHED_THE_CRITERIA_YOU))
self.mw.moveToState("overview")
QDialog.accept(self)
def reject(self) -> None:
if self.created_custom_study:
# set the original deck back to current
2020-01-16 03:45:48 +01:00
self.mw.col.decks.select(self.deck["id"])
# fixme: clean up the empty custom study deck
QDialog.reject(self)
def _getTags(self):
from aqt.taglimit import TagLimit
2019-12-23 01:34:10 +01:00
2020-05-11 17:09:22 +02:00
return TagLimit(self.mw, self).tags