anki-word-query/addons21/fastwq/context.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/>.

import json
import os
from aqt import mw
from anki.hooks import runHook
from .constants import VERSION
from .utils import get_icon


__all__ = ['APP_ICON', 'config']


APP_ICON = get_icon('wqicon.png')       #Addon Icon


class Config(object):

    """
    Addon Config
    """

    _CONFIG_FILENAME = 'fastwqcfg.json'     #Config File Path

    def __init__(self, window):
        self.path = u'_' + self._CONFIG_FILENAME
        self.window = window
        self.version = '0'
        self.data = None
        self.read()

    @property
    def pmname(self):
        return self.window.pm.name

    def update(self, data):
        """
        Update && Save
        """
        data['version'] = VERSION
        data['%s_last' % self.pmname] = data.get('last_model', self.last_model_id)
        self.data.update(data)
        with open(self.path, 'w', encoding='utf-8') as f:
            json.dump(self.data, f, indent=4, sort_keys=True, ensure_ascii=False)
            f.close()
        runHook('config.update')

    def read(self):
        """
        Load from config file
        """
        if self.data:
            return self.data
        try:
            path = self.path if os.path.exists(self.path) else u'.' + self._CONFIG_FILENAME
            with open(path, 'r', encoding="utf-8") as f:
                self.data = json.load(f)
                f.close()
            if not os.path.exists(self.path):
                self.update(self.data)
        except:
            self.data = dict()

    def get_maps(self, model_id):
        """
        Query fileds map
        """
        return self.data.get(str(model_id), list())

    @property
    def last_model_id(self):
        return self.data.get('%s_last' % self.pmname, 0)

    @property
    def dirs(self):
        return self.data.get('dirs', list())

    @property
    def dicts(self):
        return self.data.get('dicts', dict())

    @property
    def use_filename(self):
        return self.data.get('use_filename', True)

    @property
    def export_media(self):
        return self.data.get('export_media', False)

    @property
    def force_update(self):
        return self.data.get('force_update', False)
    
    @property
    def ignore_mdx_wordcase(self):
        return self.data.get('ignore_mdx_wordcase', False)

    @property
    def thread_number(self):
        """
        Query Thread Number
        """
        return self.data.get('thread_number', 16)

    @property
    def last_folder(self):
        """
        last file dialog open path 
        """
        return self.data.get('last_folder', '')

    @property
    def ignore_accents(self):
        '''ignore accents of field in querying'''
        return self.data.get('ignore_accents', False)

    @property
    def auto_update(self):
        '''auto check new version'''
        return self.data.get('auto_update', True)

config = Config(mw)