mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-07-09 03:04:10 -04:00
...
This commit is contained in:
parent
1f2aa8a55b
commit
158c33481e
@ -111,15 +111,16 @@ class Matcher(object):
|
|||||||
raise Exception('Failed to score items: %s' % error)
|
raise Exception('Failed to score items: %s' % error)
|
||||||
items = sorted(((-scores[i], item, positions[i]) for i, item in enumerate(self.items)),
|
items = sorted(((-scores[i], item, positions[i]) for i, item in enumerate(self.items)),
|
||||||
key=itemgetter(0))
|
key=itemgetter(0))
|
||||||
return OrderedDict(x[1:] for x in items)
|
return OrderedDict(x[1:] for x in filter(itemgetter(0), items))
|
||||||
|
|
||||||
def get_items_from_dir(basedir):
|
def get_items_from_dir(basedir, acceptq=lambda x: True):
|
||||||
if isinstance(basedir, bytes):
|
if isinstance(basedir, bytes):
|
||||||
basedir = basedir.decode(filesystem_encoding)
|
basedir = basedir.decode(filesystem_encoding)
|
||||||
relsep = os.sep != '/'
|
relsep = os.sep != '/'
|
||||||
for dirpath, dirnames, filenames in os.walk(basedir):
|
for dirpath, dirnames, filenames in os.walk(basedir):
|
||||||
for f in filenames:
|
for f in filenames:
|
||||||
x = os.path.join(dirpath, f)
|
x = os.path.join(dirpath, f)
|
||||||
|
if acceptq(x):
|
||||||
x = os.path.relpath(x, basedir)
|
x = os.path.relpath(x, basedir)
|
||||||
if relsep:
|
if relsep:
|
||||||
x = x.replace(os.sep, '/')
|
x = x.replace(os.sep, '/')
|
||||||
|
Loading…
x
Reference in New Issue
Block a user