Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Allow overriding prefs used when constructing tags model

Browse Source
This commit is contained in:
Kovid Goyal 2015-11-22 10:42:21 +05:30
parent b9e071e72b
commit 3387169dca
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/calibre/gui2/tag_browser/model.py
View File

@ -210,8 +210,9 @@ class TagsModel(QAbstractItemModel): # {{{
drag_drop_finished = pyqtSignal(object) drag_drop_finished = pyqtSignal(object)
user_categories_edited = pyqtSignal(object, object) user_categories_edited = pyqtSignal(object, object)
def __init__(self, parent): def __init__(self, parent, prefs=gprefs):
QAbstractItemModel.__init__(self, parent) QAbstractItemModel.__init__(self, parent)
self.prefs = prefs
self.node_map = {} self.node_map = {}
self.category_nodes = [] self.category_nodes = []
iconmap = {} iconmap = {}
@ -219,7 +220,7 @@ class TagsModel(QAbstractItemModel): # {{{
iconmap[key] = QIcon(I(category_icon_map[key])) iconmap[key] = QIcon(I(category_icon_map[key]))
self.category_icon_map = TagsIcons(iconmap) self.category_icon_map = TagsIcons(iconmap)
self.category_custom_icons = dict() self.category_custom_icons = dict()
for k, v in gprefs['tags_browser_category_icons'].iteritems(): for k, v in self.prefs['tags_browser_category_icons'].iteritems():
icon = QIcon(os.path.join(config_dir, 'tb_icons', v)) icon = QIcon(os.path.join(config_dir, 'tb_icons', v))
if len(icon.availableSizes()) > 0: if len(icon.availableSizes()) > 0:
self.category_custom_icons[k] = icon self.category_custom_icons[k] = icon
@ -242,7 +243,7 @@ class TagsModel(QAbstractItemModel): # {{{
return QObject.parent(self) return QObject.parent(self)
def set_custom_category_icon(self, key, path): def set_custom_category_icon(self, key, path):
d = gprefs['tags_browser_category_icons'] d = self.prefs['tags_browser_category_icons']
if path: if path:
d[key] = path d[key] = path
self.category_custom_icons[key] = QIcon(os.path.join(config_dir, self.category_custom_icons[key] = QIcon(os.path.join(config_dir,
@ -256,13 +257,13 @@ class TagsModel(QAbstractItemModel): # {{{
pass pass
del d[key] del d[key]
del self.category_custom_icons[key] del self.category_custom_icons[key]
gprefs['tags_browser_category_icons'] = d self.prefs['tags_browser_category_icons'] = d
def reread_collapse_model(self, state_map, rebuild=True): def reread_collapse_model(self, state_map, rebuild=True):
if gprefs['tags_browser_collapse_at'] == 0: if self.prefs['tags_browser_collapse_at'] == 0:
self.collapse_model = 'disable' self.collapse_model = 'disable'
else: else:
self.collapse_model = gprefs['tags_browser_partition_method'] self.collapse_model = self.prefs['tags_browser_partition_method']
if rebuild: if rebuild:
self.rebuild_node_tree(state_map) self.rebuild_node_tree(state_map)
@ -388,7 +389,7 @@ class TagsModel(QAbstractItemModel): # {{{
traceback.print_stack() traceback.print_stack()
return return
collapse = gprefs['tags_browser_collapse_at'] collapse = self.prefs['tags_browser_collapse_at']
collapse_model = self.collapse_model collapse_model = self.collapse_model
if collapse == 0: if collapse == 0:
collapse_model = 'disable' collapse_model = 'disable'
@ -414,7 +415,7 @@ class TagsModel(QAbstractItemModel): # {{{
is_gst = category.is_gst is_gst = category.is_gst
if key not in data: if key not in data:
return return
if key in gprefs['tag_browser_dont_collapse']: if key in self.prefs['tag_browser_dont_collapse']:
collapse_model = 'disable' collapse_model = 'disable'
cat_len = len(data[key]) cat_len = len(data[key])
if cat_len <= 0: if cat_len <= 0: