f637ac957d
- notetypes are fetched from the DB as needed, and cached in Python - handle note type changes in the backend. Multiple operations can now be performed in one go, but this is not currently exposed in the GUI. - extra methods to grab sorted note type names quickly, and fetch by name - col.models.save() without a provided notetype is now a no-op - note loading/saving handled in the backend - notes with no valid cards can now be added - templates can now be deleted even if they would previously orphan notes a number of fixmes have been left in notes.py and models.py
234 lines
7.7 KiB
Python
234 lines
7.7 KiB
Python
# Copyright: Ankitects Pty Ltd and contributors
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
import collections
|
|
import re
|
|
from operator import itemgetter
|
|
from typing import Optional
|
|
from typing import List
|
|
|
|
import aqt.clayout
|
|
from anki import stdmodels
|
|
from anki.lang import _, ngettext
|
|
from anki.rsbackend import pb
|
|
from aqt import AnkiQt, gui_hooks
|
|
from aqt.qt import *
|
|
from aqt.utils import (
|
|
askUser,
|
|
getText,
|
|
maybeHideClose,
|
|
openHelp,
|
|
restoreGeom,
|
|
saveGeom,
|
|
showInfo,
|
|
)
|
|
|
|
|
|
class Models(QDialog):
|
|
def __init__(self, mw: AnkiQt, parent=None, fromMain=False):
|
|
self.mw = mw
|
|
parent = parent or mw
|
|
self.fromMain = fromMain
|
|
QDialog.__init__(self, parent, Qt.Window)
|
|
self.col = mw.col
|
|
assert self.col
|
|
self.mm = self.col.models
|
|
self.mw.checkpoint(_("Note Types"))
|
|
self.form = aqt.forms.models.Ui_Dialog()
|
|
self.form.setupUi(self)
|
|
qconnect(self.form.buttonBox.helpRequested, lambda: openHelp("notetypes"))
|
|
self.models: List[pb.NoteTypeNameIDUseCount] = []
|
|
self.setupModels()
|
|
restoreGeom(self, "models")
|
|
self.exec_()
|
|
|
|
# Models
|
|
##########################################################################
|
|
|
|
def setupModels(self):
|
|
self.model = None
|
|
f = self.form
|
|
box = f.buttonBox
|
|
t = QDialogButtonBox.ActionRole
|
|
b = box.addButton(_("Add"), t)
|
|
qconnect(b.clicked, self.onAdd)
|
|
b = box.addButton(_("Rename"), t)
|
|
qconnect(b.clicked, self.onRename)
|
|
b = box.addButton(_("Delete"), t)
|
|
qconnect(b.clicked, self.onDelete)
|
|
if self.fromMain:
|
|
b = box.addButton(_("Fields..."), t)
|
|
qconnect(b.clicked, self.onFields)
|
|
b = box.addButton(_("Cards..."), t)
|
|
qconnect(b.clicked, self.onCards)
|
|
b = box.addButton(_("Options..."), t)
|
|
qconnect(b.clicked, self.onAdvanced)
|
|
qconnect(f.modelsList.currentRowChanged, self.modelChanged)
|
|
qconnect(f.modelsList.itemDoubleClicked, self.onRename)
|
|
self.updateModelsList()
|
|
f.modelsList.setCurrentRow(0)
|
|
maybeHideClose(box)
|
|
|
|
def onRename(self):
|
|
txt = getText(_("New name:"), default=self.model["name"])
|
|
if txt[1] and txt[0]:
|
|
self.model["name"] = txt[0]
|
|
self.mm.save(self.model, updateReqs=False)
|
|
self.updateModelsList()
|
|
|
|
def updateModelsList(self):
|
|
row = self.form.modelsList.currentRow()
|
|
if row == -1:
|
|
row = 0
|
|
self.models = self.col.models.all_use_counts()
|
|
self.form.modelsList.clear()
|
|
for m in self.models:
|
|
mUse = m.use_count
|
|
mUse = ngettext("%d note", "%d notes", mUse) % mUse
|
|
item = QListWidgetItem("%s [%s]" % (m.name, mUse))
|
|
self.form.modelsList.addItem(item)
|
|
self.form.modelsList.setCurrentRow(row)
|
|
|
|
def modelChanged(self):
|
|
if self.model:
|
|
self.saveModel()
|
|
idx = self.form.modelsList.currentRow()
|
|
self.model = self.col.models.get(self.models[idx].id)
|
|
|
|
def onAdd(self):
|
|
m = AddModel(self.mw, self).get()
|
|
if m:
|
|
txt = getText(_("Name:"), default=m["name"])[0]
|
|
if txt:
|
|
m["name"] = txt
|
|
self.mm.ensureNameUnique(m)
|
|
self.mm.save(m)
|
|
self.updateModelsList()
|
|
|
|
def onDelete(self):
|
|
if len(self.models) < 2:
|
|
showInfo(_("Please add another note type first."), parent=self)
|
|
return
|
|
if self.mm.useCount(self.model):
|
|
msg = _("Delete this note type and all its cards?")
|
|
else:
|
|
msg = _("Delete this unused note type?")
|
|
if not askUser(msg, parent=self):
|
|
return
|
|
self.mm.rem(self.model)
|
|
self.model = None
|
|
self.updateModelsList()
|
|
|
|
def onAdvanced(self):
|
|
d = QDialog(self)
|
|
frm = aqt.forms.modelopts.Ui_Dialog()
|
|
frm.setupUi(d)
|
|
frm.latexsvg.setChecked(self.model.get("latexsvg", False))
|
|
frm.latexHeader.setText(self.model["latexPre"])
|
|
frm.latexFooter.setText(self.model["latexPost"])
|
|
d.setWindowTitle(_("Options for %s") % self.model["name"])
|
|
qconnect(frm.buttonBox.helpRequested, lambda: openHelp("latex"))
|
|
restoreGeom(d, "modelopts")
|
|
gui_hooks.models_advanced_will_show(d)
|
|
d.exec_()
|
|
saveGeom(d, "modelopts")
|
|
self.model["latexsvg"] = frm.latexsvg.isChecked()
|
|
self.model["latexPre"] = str(frm.latexHeader.toPlainText())
|
|
self.model["latexPost"] = str(frm.latexFooter.toPlainText())
|
|
|
|
def saveModel(self):
|
|
self.mm.save(self.model, updateReqs=False)
|
|
|
|
def _tmpNote(self):
|
|
self.mm.setCurrent(self.model)
|
|
n = self.col.newNote(forDeck=False)
|
|
field_names = list(n.keys())
|
|
for name in field_names:
|
|
n[name] = "(" + name + ")"
|
|
|
|
cloze_re = re.compile(r"{{(?:[^}:]*:)*cloze:(?:[^}:]*:)*([^}]+)}}")
|
|
q_template = self.model["tmpls"][0]["qfmt"]
|
|
a_template = self.model["tmpls"][0]["afmt"]
|
|
|
|
used_cloze_fields = []
|
|
used_cloze_fields.extend(cloze_re.findall(q_template))
|
|
used_cloze_fields.extend(cloze_re.findall(a_template))
|
|
for field in used_cloze_fields:
|
|
if field in field_names:
|
|
n[field] = f"{field}: " + _("This is a {{c1::sample}} cloze deletion.")
|
|
|
|
return n
|
|
|
|
def onFields(self):
|
|
from aqt.fields import FieldDialog
|
|
|
|
n = self._tmpNote()
|
|
FieldDialog(self.mw, n, parent=self)
|
|
|
|
def onCards(self):
|
|
from aqt.clayout import CardLayout
|
|
|
|
n = self._tmpNote()
|
|
CardLayout(self.mw, n, ord=0, parent=self, addMode=True)
|
|
|
|
# Cleanup
|
|
##########################################################################
|
|
|
|
# need to flush model on change or reject
|
|
|
|
def reject(self):
|
|
self.saveModel()
|
|
self.mw.reset()
|
|
saveGeom(self, "models")
|
|
QDialog.reject(self)
|
|
|
|
|
|
class AddModel(QDialog):
|
|
def __init__(self, mw: AnkiQt, parent: Optional[QWidget] = None):
|
|
self.parent_ = parent or mw
|
|
self.mw = mw
|
|
self.col = mw.col
|
|
QDialog.__init__(self, self.parent_, Qt.Window)
|
|
self.model = None
|
|
self.dialog = aqt.forms.addmodel.Ui_Dialog()
|
|
self.dialog.setupUi(self)
|
|
# standard models
|
|
self.models = []
|
|
for (name, func) in stdmodels.models:
|
|
if isinstance(name, collections.Callable): # type: ignore
|
|
name = name() # type: ignore
|
|
item = QListWidgetItem(_("Add: %s") % name)
|
|
self.dialog.models.addItem(item)
|
|
self.models.append((True, func))
|
|
# add copies
|
|
for m in sorted(self.col.models.all(), key=itemgetter("name")):
|
|
item = QListWidgetItem(_("Clone: %s") % m["name"])
|
|
self.dialog.models.addItem(item)
|
|
self.models.append((False, m))
|
|
self.dialog.models.setCurrentRow(0)
|
|
# the list widget will swallow the enter key
|
|
s = QShortcut(QKeySequence("Return"), self)
|
|
qconnect(s.activated, self.accept)
|
|
# help
|
|
qconnect(self.dialog.buttonBox.helpRequested, self.onHelp)
|
|
|
|
def get(self):
|
|
self.exec_()
|
|
return self.model
|
|
|
|
def reject(self):
|
|
QDialog.reject(self)
|
|
|
|
def accept(self):
|
|
(isStd, model) = self.models[self.dialog.models.currentRow()]
|
|
if isStd:
|
|
# create
|
|
self.model = model(self.col)
|
|
else:
|
|
# add copy to deck
|
|
self.model = self.mw.col.models.copy(model)
|
|
self.mw.col.models.setCurrent(self.model)
|
|
QDialog.accept(self)
|
|
|
|
def onHelp(self):
|
|
openHelp("notetypes")
|