def __init__(self,
mw,
names=None,
accept=None,
title=None,
help="studydeck",
current=None,
cancel=True,
parent=None,
dyn=False,
buttons=None,
geomKey="default"):
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
addHook('reset', 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)
b.clicked.connect(self.onAddDeck)
if title:
self.setWindowTitle(title)
if not names:
names = sorted(self.mw.col.decks.allNames(dyn=dyn))
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)
self.form.buttonBox.helpRequested.connect(lambda: openHelp(help))
self.form.filter.textEdited.connect(self.redraw)
self.form.list.itemDoubleClicked.connect(self.accept)
self.show()
# redraw after show so position at center correct
self.redraw("", current)
self.exec_()
class TagDialogExtended__BasicOrTagEdit(QDialog):
def __init__(self, parent, tags, alltags):
QDialog.__init__(self, parent,
Qt.WindowType.Window) # super().__init__(parent)
self.basic_mode = gc("dialog type: basic_but_quick")
self.parent = parent
self.alltags = alltags
self.gridLayout = QGridLayout(self)
self.gridLayout.setObjectName("gridLayout")
self.label = QLabel("Edit tags:")
self.gridLayout.addWidget(self.label, 0, 0, 1, 1)
self.verticalLayout = QVBoxLayout()
self.verticalLayout.setObjectName("verticalLayout")
self.gridLayout.addLayout(self.verticalLayout, 1, 0, 1, 1)
self.buttonBox = QDialogButtonBox(self)
self.buttonBox.setOrientation(Qt.Orientation.Horizontal)
self.buttonBox.setStandardButtons(
QDialogButtonBox.StandardButton.Cancel
| QDialogButtonBox.StandardButton.Ok)
self.shortcut = QShortcut(QKeySequence("Ctrl+Return"), self)
self.shortcut.activated.connect(self.accept)
self.helpButton = QPushButton(
"add empty line", clicked=lambda: self.maybe_add_line(force=True))
self.buttonBox.addButton(self.helpButton,
QDialogButtonBox.ButtonRole.HelpRole)
self.filterbutton = QPushButton("edit tag for current line",
clicked=self.tagselector)
self.buttonBox.addButton(self.filterbutton,
QDialogButtonBox.ButtonRole.ResetRole)
self.searchButton = QPushButton(
"search", clicked=lambda: self.do_browser_search(extra_search=""))
self.buttonBox.addButton(self.searchButton,
QDialogButtonBox.ButtonRole.ResetRole)
self.gridLayout.addWidget(self.buttonBox, 2, 0, 1, 1)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
self.setWindowTitle("Anki - Edit Tags")
originalheight = self.height()
restoreGeom(self, "TagDialogExtended")
self.resize(self.width(), originalheight)
if not tags:
tags = [
"",
]
else:
tags.append("")
self.line_list = []
for t in tags:
self.maybe_add_line(t)
self.cut = gc("in tag lines dialog: open filterdialog for single tag")
if self.cut:
self.filterbutton.setToolTip('shortcut: {}'.format(self.cut))
self.selkey = QShortcut(QKeySequence(self.cut), self)
self.selkey.activated.connect(self.tagselector)
self.browser_scut = gc("in tag lines dialog: search browser for tag")
if self.browser_scut:
self.searchButton.setToolTip('shortcut: {}'.format(
self.browser_scut))
self.browser_scut_key = QShortcut(QKeySequence(self.browser_scut),
self)
self.browser_scut_key.activated.connect(
lambda: self.do_browser_search(extra_search=""))
# don't also set Ctrl+t,a/gc("editor: show filterdialog to add single tag") for
# self.tagselector: What if the user has already set them to the same etc. I'd have
# to do a lot of checking
self.addnl = gc("in tag lines dialog: insert additional line")
if self.addnl:
self.helpButton.setToolTip('shortcut: {}'.format(self.addnl))
self.addnlscut = QShortcut(QKeySequence(self.addnl), self)
self.addnlscut.activated.connect(
lambda: self.maybe_add_line(force=True))
def current_tags_list(self):
return [t.text() for t in self.line_list if t]
def do_browser_search(self, extra_search=""):
# Use the current line's text or the last line if the current one is an empty line
note_tags = self.current_tags_list()
searched_tag = shared_variables.focused_line.text() or (
note_tags[-1] if len(note_tags) > 0 else "")
if searched_tag:
browser = dialogs.open('Browser', mw)
browser.setFilter('tag:"{}*" {}'.format(searched_tag,
extra_search))
self.accept()
else:
tooltip("empty tag was selected for search")
def tagselector(self):
text = shared_variables.focused_line.text()
d = FilterDialog(parent=self,
values=self.alltags,
allownew=True,
prefill=text)
if d.exec():
shared_variables.focused_line.setText(d.selkey)
else:
shared_variables.focused_line.setFocus()
def change_focus_by_one(self, Down=True):
for index, edit in enumerate(self.line_list):
if edit == shared_variables.focused_line:
if Down:
if index == len(
self.line_list) - 1: # if in last line go up
self.line_list[0].setFocus()
break
else:
newidx = index + 1
self.line_list[newidx].setFocus()
break
else: # go up
if index == 0: # if in last line go up
newidx = len(self.line_list) - 1
self.line_list[newidx].setFocus()
break
else:
self.line_list[index - 1].setFocus()
break
def maybe_add_line(self, tag="", force=False):
if self.line_list and not self.line_list[-1].text(
) and not force: # last lineedit is empty:
self.line_list[-1].setFocus()
self.line_list[-1].setText(tag)
else:
if self.basic_mode:
te = MyBasicEdit(self)
te.setText(tag)
self.verticalLayout.addWidget(te)
te.setFocus()
self.line_list.append(te)
else:
te = MyTagEdit(self)
te.setCol(mw.col)
te.setText(tag)
self.verticalLayout.addWidget(te)
te.hideCompleter()
te.setFocus()
self.line_list.append(te)
def accept(self):
self.tagstring = ""
for t in self.line_list:
if not self.basic_mode:
t.hideCompleter()
text = t.text()
if text:
self.tagstring += text + " "
saveGeom(self, "TagDialogExtended")
QDialog.accept(self)
def reject(self):
saveGeom(self, "TagDialogExtended")
QDialog.reject(self)
class DeckChooser(QHBoxLayout):
def __init__(self, mw, widget, label=True, start=None):
QHBoxLayout.__init__(self)
self.widget = widget
self.mw = mw
self.deck = mw.col
self.label = label
self.setContentsMargins(0,0,0,0)
self.setSpacing(8)
self.setupDecks()
self.widget.setLayout(self)
addHook('currentModelChanged', self.onModelChange)
def setupDecks(self):
if self.label:
self.deckLabel = QLabel(_("Deck"))
self.addWidget(self.deckLabel)
# decks box
self.deck = QPushButton(clicked=self.onDeckChange)
self.deck.setToolTip(shortcut(_("Target Deck (Ctrl+D)")))
s = QShortcut(QKeySequence(_("Ctrl+D")), self.widget, activated=self.onDeckChange)
self.addWidget(self.deck)
# starting label
if self.mw.col.conf.get("addToCur", True):
col = self.mw.col
did = col.conf['curDeck']
if col.decks.isDyn(did):
# if they're reviewing, try default to current card
c = self.mw.reviewer.card
if self.mw.state == "review" and c:
if not c.odid:
did = c.did
else:
did = c.odid
else:
did = 1
self.setDeckName(self.mw.col.decks.nameOrNone(
did) or _("Default"))
else:
self.setDeckName(self.mw.col.decks.nameOrNone(
self.mw.col.models.current()['did']) or _("Default"))
# layout
sizePolicy = QSizePolicy(
QSizePolicy.Policy(7),
QSizePolicy.Policy(0))
self.deck.setSizePolicy(sizePolicy)
def show(self):
self.widget.show()
def hide(self):
self.widget.hide()
def cleanup(self):
remHook('currentModelChanged', self.onModelChange)
def onModelChange(self):
if not self.mw.col.conf.get("addToCur", True):
self.setDeckName(self.mw.col.decks.nameOrNone(
self.mw.col.models.current()['did']) or _("Default"))
def onDeckChange(self):
from aqt.studydeck import StudyDeck
current = self.deckName()
ret = StudyDeck(
self.mw, current=current, accept=_("Choose"),
title=_("Choose Deck"), help="addingnotes",
cancel=False, parent=self.widget, geomKey="selectDeck")
if ret.name:
self.setDeckName(ret.name)
def setDeckName(self, name):
self.deck.setText(name.replace("&", "&&"))
self._deckName = name
def deckName(self):
return self._deckName
def selectedId(self):
# save deck name
name = self.deckName()
if not name.strip():
did = 1
else:
did = self.mw.col.decks.id(name)
return did
class ModelChooser(QHBoxLayout):
def __init__(self, mw, widget, label=True):
QHBoxLayout.__init__(self)
self.widget = widget
self.mw = mw
self.deck = mw.col
self.label = label
self.setContentsMargins(0, 0, 0, 0)
self.setSpacing(8)
self.setupModels()
addHook('reset', self.onReset)
self.widget.setLayout(self)
def setupModels(self):
if self.label:
self.modelLabel = QLabel(_("Type"))
self.addWidget(self.modelLabel)
# models box
self.models = QPushButton()
#self.models.setStyleSheet("* { text-align: left; }")
self.models.setToolTip(shortcut(_("Change Note Type (Ctrl+N)")))
s = QShortcut(QKeySequence(_("Ctrl+N")),
self.widget,
activated=self.onModelChange)
self.models.setAutoDefault(False)
self.addWidget(self.models)
self.models.clicked.connect(self.onModelChange)
# layout
sizePolicy = QSizePolicy(QSizePolicy.Policy(7), QSizePolicy.Policy(0))
self.models.setSizePolicy(sizePolicy)
self.updateModels()
def cleanup(self):
remHook('reset', self.onReset)
def onReset(self):
self.updateModels()
def show(self):
self.widget.show()
def hide(self):
self.widget.hide()
def onEdit(self):
import aqt.models
aqt.models.Models(self.mw, self.widget)
def onModelChange(self):
from aqt.studydeck import StudyDeck
current = self.deck.models.current()['name']
# edit button
edit = QPushButton(_("Manage"), clicked=self.onEdit)
def nameFunc():
return sorted(self.deck.models.allNames())
ret = StudyDeck(self.mw,
names=nameFunc,
accept=_("Choose"),
title=_("Choose Note Type"),
help="_notes",
current=current,
parent=self.widget,
buttons=[edit],
cancel=True,
geomKey="selectModel")
if not ret.name:
return
m = self.deck.models.byName(ret.name)
self.deck.conf['curModel'] = m['id']
cdeck = self.deck.decks.current()
cdeck['mid'] = m['id']
self.deck.decks.save(cdeck)
runHook("currentModelChanged")
self.mw.reset()
def updateModels(self):
self.models.setText(self.deck.models.current()['name'])
