def getSubMenu(menu, subMenuName):
for a in menu.actions():
if subMenuName == a.text():
return a.menu()
subMenu = QMenu(subMenuName, menu)
menu.addMenu(subMenu)
return subMenu
def addMenu(name):
if not hasattr(mw, 'customMenus'):
mw.customMenus = {}
if name not in mw.customMenus:
menu = QMenu('&' + name, mw)
mw.customMenus[name] = menu
mw.form.menubar.insertMenu(mw.form.menuTools.menuAction(),
mw.customMenus[name])
def getSubMenu(menu, subMenuName):
subMenu = None
for a in menu.actions():
if subMenuName == a.text():
subMenu = a.menu()
break
if not subMenu:
subMenu = QMenu(subMenuName, menu)
menu.addMenu(subMenu)
return subMenu
def init_menu(self):
# init actions
if not self.main_menu:
self.main_menu = QMenu(
_trans("AnKindle") +
u" +" if self.ext_unlocked else _trans("AnKindle"),
mw.form.menuTools,
)
mw.form.menuTools.addMenu(self.main_menu)
self.action_show_vocab_dialog = QAction(
_trans("SHOW VOCAB IMPORT"), self.main_menu)
self.action_show_vocab_dialog.triggered.connect(
self.on_show_vocab_dialog)
self.action_show_vocab_dialog.setShortcut("CTRL+K")
self.main_menu.addAction(self.action_show_vocab_dialog)
if self.ext_available:
self.action_show_clipping_dialog = QAction(
_trans("SHOW CLIPPING IMPORT"), self.main_menu)
self.action_show_clipping_dialog.triggered.connect(
self.on_show_clipping_dialog)
self.action_show_clipping_dialog.setShortcut("CTRL+L")
self.main_menu.addAction(self.action_show_clipping_dialog)
self.main_menu.addSeparator()
if self.ext_available:
if not self.ext_unlocked:
action_upgrade = QAction(_trans("UPGRADE TO PLUS"),
self.main_menu)
action_upgrade.triggered.connect(
partial(AnKindlePlus.call_purchase_dialog, mw))
self.main_menu.addAction(action_upgrade)
if self.sn_register_dlg:
action_upgrade = QAction(_trans("REGISTER PLUS"),
self.main_menu)
AnKindlePlus._bind_qt_slots(action_upgrade.triggered,
self.on_show_enter_sn_dialog)
self.main_menu.addAction(action_upgrade)
class AnKindleAddon:
def __init__(self):
self.on_start()
# variables
self.main_menu = None
self.action_show_vocab_dialog = None
self.action_show_clipping_dialog = None
# self.main_menu_action = None
if not self.avl_col_model_names:
importFile(mw, DEFAULT_TEMPLATE)
if self.ext_available and not self.ext_unlocked:
self.sn_register_dlg = AnKindlePlus.SNReg(mw, "ANKINDLE_PLUS")
else:
self.sn_register_dlg = None
def perform_hooks(self, func):
# func('reviewCleanup', self.on_review_cleanup)
func('profileLoaded', self.on_profile_loaded)
# func('afterStateChange', self.after_anki_state_change)
def on_profile_loaded(self):
self.init_menu()
def on_start(self):
if self.ext_available:
AnKindlePlus.start_AnKindle_plus()
def init_menu(self):
# init actions
if not self.main_menu:
self.main_menu = QMenu(
_trans("AnKindle") +
u" +" if self.ext_unlocked else _trans("AnKindle"),
mw.form.menuTools,
)
mw.form.menuTools.addMenu(self.main_menu)
self.action_show_vocab_dialog = QAction(
_trans("SHOW VOCAB IMPORT"), self.main_menu)
self.action_show_vocab_dialog.triggered.connect(
self.on_show_vocab_dialog)
self.action_show_vocab_dialog.setShortcut("CTRL+K")
self.main_menu.addAction(self.action_show_vocab_dialog)
if self.ext_available:
self.action_show_clipping_dialog = QAction(
_trans("SHOW CLIPPING IMPORT"), self.main_menu)
self.action_show_clipping_dialog.triggered.connect(
self.on_show_clipping_dialog)
self.action_show_clipping_dialog.setShortcut("CTRL+L")
self.main_menu.addAction(self.action_show_clipping_dialog)
self.main_menu.addSeparator()
if self.ext_available:
if not self.ext_unlocked:
action_upgrade = QAction(_trans("UPGRADE TO PLUS"),
self.main_menu)
action_upgrade.triggered.connect(
partial(AnKindlePlus.call_purchase_dialog, mw))
self.main_menu.addAction(action_upgrade)
if self.sn_register_dlg:
action_upgrade = QAction(_trans("REGISTER PLUS"),
self.main_menu)
AnKindlePlus._bind_qt_slots(action_upgrade.triggered,
self.on_show_enter_sn_dialog)
self.main_menu.addAction(action_upgrade)
@property
def ext_available(self):
return _try_ext_module()
@property
def ext_unlocked(self):
if self.ext_available:
return AnKindlePlus.KuangKuang.Unlocked()
return False
def on_show_enter_sn_dialog(self, *args):
self.sn_register_dlg.exec_()
def on_show_clipping_dialog(self):
mw.onAddCard()
def on_show_vocab_dialog(self):
self.vocab_dlg = Window(
mw,
self.avl_col_model_names,
self.avl_decks,
)
if self.ext_unlocked:
title = "{} Plus - {}".format(_trans("AnKindle"), __version__)
if AnKindlePlus.BETA_VERSION:
title += " (BETA{})".format(
AnKindlePlus.BETA_VERSION_TEST_INDEX)
self.vocab_dlg.setWindowTitle(title)
else:
self.vocab_dlg.setWindowTitle("{} - {}".format(
_trans("AnKindle"), __version__))
self.vocab_dlg.exec_()
def avl_col_model_names(self):
_ = []
for mid, m_values in self.collection.models.models.items():
if not set([f.lower() for f in MUST_IMPLEMENT_FIELDS]).difference(
set([f[u'name'] for f in m_values[u'flds']])):
_.append(mid)
return [v for k, v in self.collection.models.models.items() if k in _]
def avl_decks(self):
_ = []
for did, d_values in self.collection.decks.decks.items():
_.append(did)
return [v for k, v in self.collection.decks.decks.items() if k in _]
@property
def collection(self):
"""
:rtype: _Collection
"""
return mw.col