104 lines
3.3 KiB
Python
104 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright: Ankitects Pty Ltd and contributors
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
from typing import Optional
|
|
|
|
from aqt import AnkiQt, gui_hooks
|
|
from aqt.qt import *
|
|
from aqt.utils import TR, shortcut, tr
|
|
|
|
|
|
class ModelChooser(QHBoxLayout):
|
|
def __init__(
|
|
self,
|
|
mw: AnkiQt,
|
|
widget: QWidget,
|
|
label: bool = True,
|
|
on_activated: Optional[Callable[[], None]] = None,
|
|
) -> None:
|
|
"""If provided, on_activated() will be called when the button is clicked,
|
|
and the caller can call .onModelChange() to pull up the dialog when they
|
|
are ready."""
|
|
QHBoxLayout.__init__(self)
|
|
self.widget = widget # type: ignore
|
|
self.mw = mw
|
|
self.deck = mw.col
|
|
self.label = label
|
|
if on_activated:
|
|
self.on_activated = on_activated
|
|
else:
|
|
self.on_activated = self.onModelChange
|
|
self.setContentsMargins(0, 0, 0, 0)
|
|
self.setSpacing(8)
|
|
self.setupModels()
|
|
gui_hooks.state_did_reset.append(self.onReset)
|
|
self.widget.setLayout(self)
|
|
|
|
def setupModels(self) -> None:
|
|
if self.label:
|
|
self.modelLabel = QLabel(tr(TR.NOTETYPES_TYPE))
|
|
self.addWidget(self.modelLabel)
|
|
# models box
|
|
self.models = QPushButton()
|
|
self.models.setToolTip(shortcut(tr(TR.QT_MISC_CHANGE_NOTE_TYPE_CTRLANDN)))
|
|
QShortcut(QKeySequence("Ctrl+N"), self.widget, activated=self.on_activated) # type: ignore
|
|
self.models.setAutoDefault(False)
|
|
self.addWidget(self.models)
|
|
qconnect(self.models.clicked, self.onModelChange)
|
|
# layout
|
|
sizePolicy = QSizePolicy(QSizePolicy.Policy(7), QSizePolicy.Policy(0))
|
|
self.models.setSizePolicy(sizePolicy)
|
|
self.updateModels()
|
|
|
|
def cleanup(self) -> None:
|
|
gui_hooks.state_did_reset.remove(self.onReset)
|
|
|
|
def onReset(self) -> None:
|
|
self.updateModels()
|
|
|
|
def show(self) -> None:
|
|
self.widget.show() # type: ignore
|
|
|
|
def hide(self) -> None:
|
|
self.widget.hide() # type: ignore
|
|
|
|
def onEdit(self) -> None:
|
|
import aqt.models
|
|
|
|
aqt.models.Models(self.mw, self.widget)
|
|
|
|
def onModelChange(self) -> None:
|
|
from aqt.studydeck import StudyDeck
|
|
|
|
current = self.deck.models.current()["name"]
|
|
# edit button
|
|
edit = QPushButton(tr(TR.QT_MISC_MANAGE), clicked=self.onEdit) # type: ignore
|
|
|
|
def nameFunc():
|
|
return sorted(self.deck.models.allNames())
|
|
|
|
ret = StudyDeck(
|
|
self.mw,
|
|
names=nameFunc,
|
|
accept=tr(TR.ACTIONS_CHOOSE),
|
|
title=tr(TR.QT_MISC_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)
|
|
gui_hooks.current_note_type_did_change(current)
|
|
self.mw.reset()
|
|
|
|
def updateModels(self) -> None:
|
|
self.models.setText(self.deck.models.current()["name"].replace("&", "&&"))
|