summaryrefslogtreecommitdiffstats
path: root/unit_trace/maxer.py
diff options
context:
space:
mode:
Diffstat (limited to 'unit_trace/maxer.py')
-rw-r--r--unit_trace/maxer.py20
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 @@
1###############################################################################
2# Description
3###############################################################################
4
5# Parse at most the given number of records
6
7###############################################################################
8# Public functions
9###############################################################################
10
11def maxer(stream, number):
12 for record in stream:
13 if record.record_type=="event":
14 if number > 0:
15 number -= 1
16 yield record
17 else:
18 break
19 else:
20 yield record