Python-bindings: Refactor the Context class

[babeltrace.git] / bindings / python / examples / babeltrace_and_lttng.py

diff --git a/bindings/python/examples/babeltrace_and_lttng.py b/bindings/python/examples/babeltrace_and_lttng.py
index 3a99b1fbae2af1ad3eca68b725cda34553361c04..3e7d57f53627646ec9eb7f7c622412e1402a289e 100644 (file)
--- a/bindings/python/examples/babeltrace_and_lttng.py
+++ b/bindings/python/examples/babeltrace_and_lttng.py
@@ -97,36 +97,23 @@ if ret < 0:
 
 # BABELTRACE
 
-# Create context and add trace:
-ctx = babeltrace.Context()
-ret = ctx.add_trace(trace_path + "/kernel", "ctf")
+# Create TraceCollecion and add trace:
+traces = babeltrace.TraceCollection()
+ret = traces.add_trace(trace_path + "/kernel", "ctf")
 if ret is None:
        raise BabeltraceError("Error adding trace")
 
-# Iterator setup
-bp = babeltrace.IterPos(babeltrace.SEEK_BEGIN)
-ctf_it = babeltrace.CTFReader.Iterator(ctx,bp)
-
 # Reading events from trace
 # and outputting timestamps and event names
 # in out_file
 print("Writing trace file...")
 output = open(out_file, "wt")
 
-event = ctf_it.read_event()
-while(event is not None):
+for event in traces.events:
        output.write("TS: {}, {} : {}\n".format(event.get_timestamp(),
                event.get_cycles(), event.get_name()))
 
-       # Next event
-       ret = ctf_it.next()
-       if ret < 0:
-               break
-       event = ctf_it.read_event()
-
 # Closing file
 output.close()
 
-# Destroying dynamic elements
-del ctf_it, han
 print("Done.")