summaryrefslogtreecommitdiffstats
path: root/unit_trace/maxer.py
blob: 086c6bcf2b59d258920e1568f014e8ae74c04a84 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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