Created maxer and skipper

Maxer: Allow a maximum number of records to be parsed
Skipper: Skip a given number of records at the beginning

1 files changed, 20 insertions, 0 deletions

diff --git a/unit_trace/maxer.py b/unit_trace/maxer.py
new file mode 100644
index 0000000..086c6bc
--- /dev/null
+++ b/unit_trace/maxer.py
@@ -0,0 +1,20 @@
+###############################################################################
+# Description
+###############################################################################
+
+# Parse at most the given number of records
+
+###############################################################################
+# Public functions
+###############################################################################
+
+def maxer(stream, number):
+    for record in stream:
+        if record.record_type=="event":
+            if number > 0:
+                number -= 1
+                yield record
+            else:
+                break
+        else:
+            yield record