anki-word-query/addons/fastwq/__init__.py

#-*- coding:utf-8 -*-
#
# Copyright (C) 2018 sthoo <sth201807@gmail.com>
#
# Support: Report an issue at https://github.com/sth2018/FastWordQuery/issues
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# any later version; http://www.gnu.org/copyleft/gpl.html.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.

from operator import itemgetter
from aqt import mw
from aqt.qt import *
from anki.hooks import addHook, wrap, remHook
from aqt.addcards import AddCards
from aqt.utils import showInfo, shortcut, downArrow
from .gui import show_options, show_about_dialog #, check_updates
from .query import query_from_browser, query_from_editor_fields
from .context import config, APP_ICON, Config
from .lang import _
from .service import service_pool
from .utils import get_icon


__all__ = [
    'browser_menu', 'customize_addcards', 
    'config_menu', 'context_menu' #, 'check_updates'
]


have_setup = False
my_shortcut = ''

_OK_ICON = get_icon('ok.png')
_NULL_ICON = get_icon('null.png')


def set_options_def(mid, i):
    conf = config.get_maps(mid)
    conf = {'list': [conf], 'def': 0} if isinstance(conf, list) else conf
    if conf['def'] != i:
        conf['def'] = i
        data = dict()
        data[mid] = conf
        config.update(data)
# end set_options_def

def browser_menu():
    """
    add add-on's menu to browser window
    """
    def on_setup_menus(browser):
        """
        on browser setupMenus was called
        """
        # main menu
        menu = browser.form.menubar.addMenu("FastWQ")
        # menu gen
        def init_fastwq_menu():
            try:
                menu.clear()
            except RuntimeError:
                remHook('config.update', init_fastwq_menu)
                return
            # Query Selected
            action = QAction(_('QUERY_SELECTED'), browser)
            action.triggered.connect(lambda: query_from_browser(browser))
            action.setShortcut(QKeySequence(my_shortcut))
            menu.addAction(action)
            # Options
            action = QAction(_('OPTIONS'), browser)
            def _show_options():
                model_id = -1
                for note_id in browser.selectedNotes():
                    note = browser.mw.col.getNote(note_id)
                    model_id = note.model()['id']
                    break
                show_options(browser, model_id)
            action.triggered.connect(_show_options)
            menu.addAction(action)

            # Default configs
            menu.addSeparator()
            b = False
            for m in sorted(browser.mw.col.models.all(), key=itemgetter("name")):
                conf = config.get_maps(m['id'])
                conf = {'list': [conf], 'def': 0} if isinstance(conf, list) else conf
                maps_list = conf['list']
                if len(maps_list) > 1:
                    submenu = menu.addMenu(m['name'])
                    for i, maps in enumerate(maps_list):
                        submenu.addAction(
                            _OK_ICON if i==conf['def'] else _NULL_ICON,
                            _('CONFIG_INDEX') % (i+1) if isinstance(maps, list) else maps['name'],
                            lambda mid=m['id'], i=i: set_options_def(mid, i)
                        )
                    b = True
            if b: 
                menu.addSeparator()

            # # check update
            # action = QAction(_('CHECK_UPDATE'), browser)
            # action.triggered.connect(lambda: check_updates(background=False, parent=browser))
            # menu.addAction(action)

            # About
            action = QAction(_('ABOUT'), browser)
            action.triggered.connect(lambda: show_about_dialog(browser))
            menu.addAction(action)

        # end init_fastwq_menu
        init_fastwq_menu()
        addHook('config.update', init_fastwq_menu)
    addHook('browser.setupMenus', on_setup_menus)


def customize_addcards():
    """
    add button to addcards window
    """
    def add_query_button(self):
        '''
        add a button in add card window
        '''
        bb = self.form.buttonBox
        ar = QDialogButtonBox.ActionRole
        # button
        fastwqBtn = QPushButton(_("QUERY") + u" " + downArrow())
        fastwqBtn.setShortcut(QKeySequence(my_shortcut))
        fastwqBtn.setToolTip(
            _(u"Shortcut: %s") % shortcut(my_shortcut)
        )
        bb.addButton(fastwqBtn, ar)
        # signal
        def onQuery(e):
            if isinstance(e, QMouseEvent):
                if e.buttons() & Qt.LeftButton:
                    menu = QMenu(self)
                    menu.addAction(_("ALL_FIELDS"), lambda: query_from_editor_fields(self.editor), QKeySequence(my_shortcut))
                    # default options
                    mid = self.editor.note.model()['id']
                    conf = config.get_maps(mid)
                    conf = {'list': [conf], 'def': 0} if isinstance(conf, list) else conf
                    maps_list = conf['list']
                    if len(maps_list) > 1:
                        menu.addSeparator()
                        for i, maps in enumerate(maps_list):
                            menu.addAction(
                                _OK_ICON if i==conf['def'] else _NULL_ICON,
                                _('CONFIG_INDEX') % (i+1) if isinstance(maps, list) else maps['name'],
                                lambda mid=mid, i=i: set_options_def(mid, i)
                            )
                        menu.addSeparator()
                    # end default options
                    menu.addAction(_("OPTIONS"), lambda: show_options(self, self.editor.note.model()['id']))
                    menu.exec_(fastwqBtn.mapToGlobal(QPoint(0, fastwqBtn.height())))
            else:
                query_from_editor_fields(self.editor)

        fastwqBtn.mousePressEvent = onQuery
        fastwqBtn.clicked.connect(onQuery)
    
    AddCards.setupButtons = wrap(
        AddCards.setupButtons, 
        add_query_button, 
        "after"
    )


def config_menu():
    """
    add menu to anki window menebar
    """
    action = QAction(APP_ICON, "FastWQ...", mw)
    action.triggered.connect(lambda: show_options())
    mw.form.menuTools.addAction(action)


def context_menu():
    '''mouse right click menu'''
    def on_setup_menus(web_view, menu):
        """
        add context menu to webview
        """
        if not isinstance(web_view.editor.currentField, int):
            return
        current_model_id = web_view.editor.note.model()['id']
        conf = config.get_maps(current_model_id)
        maps_list = conf if isinstance(conf, list) else conf['list']
        curr_flds = []
        names = []
        for i, maps in enumerate(maps_list):
            maps = maps if isinstance(maps, list) else maps['fields']
            for mord, m in enumerate(maps):
                if m.get('word_checked', False):
                    word_ord = mord
                    break
            if web_view.editor.currentField != word_ord:
                each = maps[web_view.editor.currentField]
                ignore = each.get('ignore', False)
                if not ignore:
                    dict_unique = each.get('dict_unique', '').strip()
                    dict_fld_ord = each.get('dict_fld_ord', -1)
                    fld_ord = each.get('fld_ord', -1)
                    if dict_unique and dict_fld_ord != -1 and fld_ord != -1:
                        s = service_pool.get(dict_unique)
                        if s and s.support:
                            name = s.title + ' :-> ' + s.fields[dict_fld_ord]
                            if name not in names:
                                names.append(name)
                                curr_flds.append({
                                    'name': name,
                                    'def': i
                                })
                        service_pool.put(s)

        submenu = menu.addMenu(_('QUERY'))
        submenu.addAction(_('ALL_FIELDS'), lambda: query_from_editor_fields(web_view.editor), QKeySequence(my_shortcut))
        if len(curr_flds) > 0:
            # quer hook method
            def query_from_editor_hook(i):
                conf = config.get_maps(current_model_id)
                maps_old_def = 0 if isinstance(conf, list) else conf.get('def', 0)
                set_options_def(current_model_id, i)
                query_from_editor_fields(
                    web_view.editor,
                    fields=[web_view.editor.currentField]
                )
                set_options_def(current_model_id, maps_old_def)
            # sub menu
            #flds_menu = submenu.addMenu(_('CURRENT_FIELDS'))
            submenu.addSeparator()
            for c in curr_flds:
                submenu.addAction(c['name'], 
                    lambda i=c['def']: query_from_editor_hook(i)
                )
            submenu.addSeparator()
        submenu.addAction(_("OPTIONS"), lambda: show_options(web_view, web_view.editor.note.model()['id']))

    addHook('EditorWebView.contextMenuEvent', on_setup_menus)