1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/test_write.py Mon Oct 03 00:17:53 2011 +0200
1.3 @@ -0,0 +1,40 @@
1.4 +#!/usr/bin/env python
1.5 +
1.6 +from simplex import *
1.7 +import sys, time
1.8 +
1.9 +class IndexWriter:
1.10 +
1.11 + "A tab-delimited file writer."
1.12 +
1.13 + def __init__(self, f):
1.14 + self.f = f
1.15 +
1.16 + def append(self, entry):
1.17 + key, pos = entry
1.18 + entry = key + [pos]
1.19 + self.f.write("\t".join(map(str, entry)) + "\n")
1.20 +
1.21 +try:
1.22 + filename, numeric, interval, index_filename = sys.argv[1:5]
1.23 + fields = map(int, sys.argv[5:])
1.24 + numeric = numeric == "true"
1.25 +except (IndexError, ValueError):
1.26 + print >>sys.stderr, "Usage: %s <filename> <numeric> <interval> <index> <field>..." % sys.argv[0]
1.27 + sys.exit(1)
1.28 +
1.29 +f = open(filename)
1.30 +fi = open(index_filename, "w")
1.31 +accessor = DelimitedRecord(fields, numeric=numeric)
1.32 +writer = IndexWriter(fi)
1.33 +
1.34 +try:
1.35 + t = time.time()
1.36 + make_index(f, accessor.get_key, int(interval), writer)
1.37 + print "Indexed in %s seconds." % (time.time() - t)
1.38 +
1.39 +finally:
1.40 + f.close()
1.41 + fi.close()
1.42 +
1.43 +# vim: tabstop=4 expandtab shiftwidth=4