Bugfix: small traces yield no events

A bug caused traces with < 100 records (the number of records
buffered and sorted) to yield no records.

1 files changed, 9 insertions, 2 deletions

diff --git a/unit_trace/trace_reader.py b/unit_trace/trace_reader.py
index 5cfbbe3..5927250 100644
--- a/unit_trace/trace_reader.py
+++ b/unit_trace/trace_reader.py
@@ -64,9 +64,16 @@ def trace_reader(files):
     # This is because records may have been recorded slightly out of order
     # This cannot guarantee records are produced in order, but it makes it
     # overwhelmingly probably.
+    # The 'try' and 'except' catches the case where there are less than 100
+    # records in a file (throwing a StopIteration) which otherwise would
+    # propogate up and cause the trace_reader generator itself to throw a
+    # StopIteration.
     for x in range(0,len(file_iter_buff)):
-        for y in range(0,100):
-            file_iter_buff[x].append(file_iters[x].next())
+        try:
+            for y in range(0,100):
+                file_iter_buff[x].append(file_iters[x].next())
+        except StopIteration:
+            pass
     for x in range(0,len(file_iter_buff)):
         file_iter_buff[x] = sorted(file_iter_buff[x],key=lambda rec: rec.when)