1 #!/usr/bin/env python 2 3 from simplex import * 4 import sys, time 5 6 try: 7 separator = sys.argv.index("--") 8 filename, numeric, interval = sys.argv[1:4] 9 fields = map(int, sys.argv[4:separator]) 10 terms = groups(sys.argv[separator+1:], len(fields)) 11 except (IndexError, ValueError): 12 print >>sys.stderr, "Usage: %s <filename> <interval> <field>... -- <term value>..." % sys.argv[0] 13 sys.exit(1) 14 15 f = open(filename) 16 converters = [(numeric == "true" and int or None) for field in fields] 17 accessor = Converted(DelimitedRecord(fields), converters) 18 reader = TextFile(f, Iterator(accessor)) 19 20 try: 21 t = time.time() 22 l = make_index(reader, int(interval)) 23 print "Indexed in %s seconds." % (time.time() - t) 24 25 # Now use the index. 26 27 for term in terms: 28 t = time.time() 29 line = find_with_index(reader, l, accessor.convert(term)) 30 if line: 31 print "Found (at %s seconds)...\n%s" % (time.time() - t, line) 32 33 finally: 34 f.close() 35 36 # vim: tabstop=4 expandtab shiftwidth=4