1.1 --- a/itermerge.py Tue Feb 08 00:08:27 2011 +0100
1.2 +++ b/itermerge.py Thu Feb 10 01:19:13 2011 +0100
1.3 @@ -29,6 +29,7 @@
1.4
1.5 def __init__(self, sequences):
1.6 self.iters = []
1.7 + self.first = None
1.8
1.9 # Prepare the underlying iterators.
1.10
1.11 @@ -43,6 +44,15 @@
1.12 next = iterator.next
1.13 self._add_next(next)
1.14
1.15 + def __getitem__(self, i):
1.16 + if i == 0:
1.17 + if self.first is None:
1.18 + value, next = self.iters[0]
1.19 + self.first = value
1.20 + return self.first
1.21 + else:
1.22 + raise IndexError, "Index %d cannot be accessed in this iterator." % i
1.23 +
1.24 def sort(self):
1.25 pass # The output should be sorted.
1.26