1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
|
#! /usr/bin/env python
import sys, os
from zeitgeist.datamodel import Event, Subject, ResultType, TimeRange
from zeitgeist.client import ZeitgeistDBusInterface
USAGE = \
"""Talis is a command line tool to search your history.
USAGE:
talis [options] <enter search terms here>
OPTIONS:
--mode=, -m= MODE
--offset=, -o= Offset into result set
--hits=, -h= Max number of hits to return
--relevancies, -r With relevancies
MODES:
LeastPopularActor
LeastPopularSubjects
LeastRecentActor
LeastRecentEvents
LeastRecentSubjects
MostPopularActor
MostPopularSubjects
MostRecentActor
MostRecentEvents
MostRecentSubjects
Relevancy (default)
"""
if len(sys.argv) <= 1:
print USAGE
raise SystemExit(1)
offset = 0
maxhits = 10
mode = 100
mode_string = "Relevancy"
query = ""
relevancy_mode = False
for i in range(1, len(sys.argv)):
if sys.argv[i].startswith("--mode=") or sys.argv[i].startswith("-m="):
mode_string = sys.argv[1].replace("--mode=", "").replace("-m=", "").strip()
if mode_string == "Relevancy":
mode = 100
else:
mode = getattr(ResultType, mode_string)
elif sys.argv[i].startswith("--offset=") or sys.argv[i].startswith("-o="):
offset = int(sys.argv[i].replace("--offset=", "").replace("-o=", "").strip())
elif sys.argv[i].startswith("--hits=") or sys.argv[i].startswith("-h="):
maxhits = int(sys.argv[i].replace("--hits=", "").replace("-h=", "").strip())
elif sys.argv[i] in ["--relevancies", "-r"]:
relevancy_mode = True
else:
query += " " + sys.argv[i]
query = query.strip()
fts = ZeitgeistDBusInterface().get_extension("Index", "index/activity")
if relevancy_mode:
results, relevancies, count = fts.SearchWithRelevancies(query, TimeRange.always(), [], 2, offset, maxhits, mode)
else:
results, count = fts.Search(query, TimeRange.always(), [], offset, maxhits, mode)
if len(results) == 0:
print "No hits for '%s'" % query
raise SystemExit(0)
print "%s-%s of %s hits for '%s'. Sorted by '%s':" % (offset+1,
len(results)+offset, count, query, mode_string)
i = offset+1
events = map(Event, results)
for event in events:
for subject in event.subjects:
if relevancy_mode:
print " %s. %s\t(%.2f)" % (i, subject.uri, relevancies[i-1])
else:
print " %s. %s" % (i, subject.uri)
i += 1
|