utils/trace_cmd: add trace_has_marker()

Add a function to check whether a trace file contains start/stop markers
(actually only check for start marker -- stop marker is assumed).

1 files changed, 9 insertions, 0 deletions

diff --git a/wa/utils/trace_cmd.py b/wa/utils/trace_cmd.py
index b308f41a..8eaf4d59 100644
--- a/wa/utils/trace_cmd.py
+++ b/wa/utils/trace_cmd.py
@@ -325,3 +325,12 @@ class TraceCmdParser(object):
                     body_parser = regex_body_parser(body_parser)
                 yield TraceCmdEvent(parser=body_parser, **match.groupdict())
 
+
+def trace_has_marker(filepath, max_lines_to_check=2000000):
+    with open(filepath) as fh:
+        for i, line in enumerate(fh):
+            if TRACE_MARKER_START in line:
+                return True
+            if i >= max_lines_to_check:
+                break
+    return False