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

Merge from trunk

Browse Source
This commit is contained in:
Charles Haley 2011-03-08 19:59:42 +00:00
commit 254e2bcca0
1 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/calibre/library/caches.py
View File

@ -6,8 +6,8 @@ __license__ = 'GPL v3'
__copyright__ = '2010, Kovid Goyal <kovid@kovidgoyal.net>' __copyright__ = '2010, Kovid Goyal <kovid@kovidgoyal.net>'
__docformat__ = 'restructuredtext en' __docformat__ = 'restructuredtext en'
import re, itertools, time, traceback, copy import re, itertools, time, traceback
from itertools import repeat from itertools import repeat, izip, imap
from datetime import timedelta from datetime import timedelta
from threading import Thread from threading import Thread
@ -777,18 +777,30 @@ class ResultCache(SearchQueryParser): # {{{
return self.search_restriction_book_count return self.search_restriction_book_count
def set_marked_ids(self, id_dict): def set_marked_ids(self, id_dict):
if isinstance (id_dict, list): '''
ids in id_dict are "marked". They can be searched for by
using the search term ``marked:true``. Pass in an empty dictionary or
set to clear marked ids.
:param id_dict: Either a dictionary mapping ids to values or a set
of ids. In the latter case, the value is set to 'true' for all ids. If
a mapping is provided, then the search can be used to search for
particular values: ``marked:value``
'''
if not hasattr(id_dict, 'items'):
# Simple list. Make it a dict of string 'true' # Simple list. Make it a dict of string 'true'
self.marked_ids_dict = dict([(id, u'true') for id in id_dict]) self.marked_ids_dict = dict.fromkeys(id_dict, u'true')
else: else:
self.marked_ids_dict = copy.copy(id_dict)
# Ensure that all the items in the dict are text # Ensure that all the items in the dict are text
for id_,val in self.marked_ids_dict.iteritems(): self.marked_ids_dict = dict(izip(id_dict.iterkeys(), imap(unicode,
self.marked_ids_dict[id_] = unicode(val) id_dict.itervalues())))
# Set the values in the cache # Set the values in the cache
marked_col = self.FIELD_MAP['marked'] marked_col = self.FIELD_MAP['marked']
for id_,val in self.marked_ids_dict.iteritems(): for r in self.iterall():
r[marked_col] = None
for id_, val in self.marked_ids_dict.iteritems():
try: try:
self._data[id_][marked_col] = val self._data[id_][marked_col] = val
except: except: