mirror of
https://github.com/ankitects/anki.git
synced 2025-09-18 22:12:21 -04:00
93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright: Damien Elmes <anki@ichi2.net>
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
from aqt.qt import *
|
|
from operator import itemgetter
|
|
from anki.hooks import addHook, remHook, runHook
|
|
from aqt.utils import isMac
|
|
import aqt
|
|
|
|
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._ignoreReset = False
|
|
self.setMargin(0)
|
|
self.setSpacing(8)
|
|
self.setupModels()
|
|
addHook('reset', self.onReset)
|
|
self.widget.setLayout(self)
|
|
|
|
def setupModels(self):
|
|
if self.label:
|
|
self.modelLabel = QLabel(_("Note Type:"))
|
|
self.addWidget(self.modelLabel)
|
|
# models dropdown
|
|
self.models = QComboBox()
|
|
s = QShortcut(QKeySequence(_("Shift+Alt+m")), self.widget)
|
|
s.connect(s, SIGNAL("activated()"),
|
|
lambda: self.models.showPopup())
|
|
self.addWidget(self.models)
|
|
self.connect(self.models, SIGNAL("activated(int)"), self.onModelChange)
|
|
# edit button
|
|
self.edit = QPushButton()
|
|
if isMac:
|
|
self.edit.setFixedWidth(24)
|
|
self.edit.setFixedHeight(21)
|
|
else:
|
|
self.edit.setFixedWidth(32)
|
|
self.edit.setIcon(QIcon(":/icons/gears.png"))
|
|
self.edit.setShortcut(_("Shift+Alt+e"))
|
|
self.edit.setToolTip(_("Customize Note Types"))
|
|
self.edit.setAutoDefault(False)
|
|
self.addWidget(self.edit)
|
|
self.connect(self.edit, SIGNAL("clicked()"), self.onEdit)
|
|
# 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):
|
|
if not self._ignoreReset:
|
|
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, idx):
|
|
model = self._models[idx]
|
|
self.deck.conf['curModel'] = model['id']
|
|
cdeck = self.deck.decks.current()
|
|
cdeck['mid'] = model['id']
|
|
self.deck.decks.save(cdeck)
|
|
self._ignoreReset = True
|
|
runHook("currentModelChanged")
|
|
self._ignoreReset = False
|
|
|
|
def updateModels(self):
|
|
self.models.clear()
|
|
self._models = sorted(self.deck.models.all(),
|
|
key=itemgetter("name"))
|
|
self.models.addItems([m['name'] for m in self._models])
|
|
cur = self.deck.models.current()
|
|
for c, m in enumerate(self._models):
|
|
if m['id'] == cur['id']:
|
|
self.models.setCurrentIndex(c)
|
|
break
|