def search(argv):
import getopt, locale, time
def usage():
print ('usage: %s [-d] [-T timeout] [-s|-Y] [-D] [-a] '
'[-c savefile] [-b basedir] [-p prefix] [-t doctype] '
'[-e encoding] [-n results] idxdir [keyword ...]') % argv[0]
sys.exit(2)
try:
(opts, args) = getopt.getopt(argv[1:], 'dT:sYDac:b:p:t:e:n:')
except getopt.GetoptError:
usage()
debug = 0
timeout = 0
stat = False
disjunctive = False
savefile = ''
basedir = ''
prefix = ''
doctype = document.PlainTextDocument
predtype = KeywordPredicate
encoding = locale.getpreferredencoding()
n = 10
for (k, v) in opts:
if k == '-d': debug += 1
elif k == '-T': timeout = int(v)
elif k == '-D': disjunctive = True
elif k == '-a': stat = True
elif k == '-Y': predtype = YomiKeywordPredicate
elif k == '-s': predtype = StrictKeywordPredicate
elif k == '-c': savefile = v
elif k == '-b': basedir = v
elif k == '-p': prefix = v
elif k == '-t': doctype = document.get_doctype(v)
elif k == '-e': encoding = v
elif k == '-n': n = int(v)
if doctype == document.EMailDocument:
predtype = EMailPredicate
t0 = time.time()
if args:
idxdir = args[0]
keywords = args[1:]
indexdb = IndexDB(idxdir, prefix)
indexdb.open()
preds = [ predtype(unicode(kw, encoding)) for kw in keywords ]
selection = Selection(indexdb, preds, disjunctive=disjunctive)
selection.set_timeout(timeout)
try:
show_results(selection, n, encoding)
except SearchTimeout:
print 'SearchTimeout.'
elif savefile:
selection = load_selection(savefile)
selection.set_timeout(timeout)
try:
show_results(selection, n, encoding)
except SearchTimeout:
print 'SearchTimeout.'
else:
usage()
if savefile:
save_selection(savefile, selection)
if stat:
print '%.2f sec, %d/%d hit' % (time.time()-t0, len(selection.found_docs), selection.narrowed)
return
