162 lines
4.3 KiB
Python
162 lines
4.3 KiB
Python
# -*- 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 anki.hooks import runHook
|
|
from aqt import mw
|
|
|
|
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, 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 Exception as e:
|
|
print('can not find config file', e)
|
|
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 cloze_str(self):
|
|
'''cloze formater string'''
|
|
tmpstr = self.data.get('cloze_str', '{{c1::%s}}')
|
|
if len(tmpstr.split('%s')) != 2:
|
|
tmpstr = '{{c1::%s}}'
|
|
return tmpstr
|
|
|
|
@property
|
|
def sound_str(self):
|
|
'''sound formater string'''
|
|
# 设置音频播放按钮大小
|
|
# <span style="width:24px;height:24px;">[sound:{0}]</span>
|
|
tmpstr = self.data.get('sound_str', u'[sound:{0}]')
|
|
if len(tmpstr.split('{0}')) != 2:
|
|
tmpstr = u'[sound:{0}]'
|
|
return tmpstr
|
|
|
|
|
|
# config = Config(mw)
|
|
# https://github.com/sth2018/FastWordQuery/issues/258
|
|
wp = mw.pm.profileFolder()
|
|
mediaPath = os.path.join(wp, "collection.media")
|
|
os.chdir(mediaPath)
|
|
config = Config(mw)
|