Add support for intervals

Can now filter by earliest record of interest and
latest record of interest, using -e and -l.

2 files changed, 40 insertions, 0 deletions

diff --git a/unit_trace/earliest.py b/unit_trace/earliest.py
new file mode 100644
index 0000000..eaf0cf8
--- /dev/null
+++ b/unit_trace/earliest.py
@@ -0,0 +1,21 @@
+###############################################################################
+# Description
+###############################################################################
+
+# Enforce earliest record
+
+###############################################################################
+# Public functions
+###############################################################################
+
+def earliest(stream, earliest):
+    for record in stream:
+        if record.record_type=="event":
+            if record.when < earliest:
+                pass
+            else:
+                yield record
+                break
+        yield record
+    for record in stream:
+        yield record
diff --git a/unit_trace/latest.py b/unit_trace/latest.py
new file mode 100644
index 0000000..4abd3a2
--- /dev/null
+++ b/unit_trace/latest.py
@@ -0,0 +1,19 @@
+###############################################################################
+# Description
+###############################################################################
+
+# Enforce latest record
+
+###############################################################################
+# Public functions
+###############################################################################
+
+def latest(stream, latest):
+    for record in stream:
+        if record.record_type=="event":
+            if record.when > latest:
+                break
+            else:
+                yield record
+        else:
+            yield record