Python bindings: Make examples PEP8 compliant

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

diff --git a/bindings/python/examples/ctf_writer.py b/bindings/python/examples/ctf_writer.py
index fd9ce1595b26a48313638700f424e2ab456d3eb7..178eed313c60ae6f3ec425f0eb52badee2127964 100644 (file)
--- a/bindings/python/examples/ctf_writer.py
+++ b/bindings/python/examples/ctf_writer.py
@@ -36,6 +36,7 @@ print("Clock offset_seconds is {}".format(clock.offset_seconds))
 print("Clock offset is {}".format(clock.offset))
 print("Clock is absolute: {}".format(clock.absolute))
 print("Clock time is {}".format(clock.time))
+print("Clock UUID is {}".format(clock.uuid))
 
 writer.add_clock(clock)
 writer.add_environment_field("Python_version", str(sys.version_info))
@@ -53,6 +54,14 @@ int32_type.signed = True
 uint16_type = CTFWriter.IntegerFieldDeclaration(16)
 uint16_type.signed = False
 
+# Add a custom uint16_t field in the stream's packet context
+packet_context_type = stream_class.packet_context_type
+print("\nFields in default packet context:")
+for field in packet_context_type.fields:
+    print(str(type(field[1])) + " " + field[0])
+packet_context_type.add_field(uint16_type, "a_custom_packet_context_field")
+stream_class.packet_context_type = packet_context_type
+
 # Create a string type
 string_type = CTFWriter.StringFieldDeclaration()
 
@@ -89,34 +98,39 @@ stream_class.add_event_class(event_class)
 stream = writer.create_stream(stream_class)
 
 for i in range(100):
-       event = CTFWriter.Event(event_class)
+    event = CTFWriter.Event(event_class)
+
+    clock.time = i * 1000
+    structure_field = event.payload("structure_field")
+    integer_field = structure_field.field("an_integer")
+    integer_field.value = i
 
-       clock.time = i * 1000
-       structure_field = event.payload("structure_field")
-       integer_field = structure_field.field("an_integer")
-       integer_field.value = i
+    string_field = structure_field.field("a_string_field")
+    string_field.value = "Test string."
 
-       string_field = structure_field.field("a_string_field")
-       string_field.value = "Test string."
+    float_field = event.payload("float_field")
+    float_field.value = float(i) + (float(i) / 100.0)
 
-       float_field = event.payload("float_field")
-       float_field.value = float(i) + (float(i) / 100.0)
+    array_field = event.payload("array_field")
+    for j in range(5):
+        element = array_field.field(j)
+        element.value = i + j
 
-       array_field = event.payload("array_field")
-       for j in range(5):
-               element = array_field.field(j)
-               element.value = i + j
+    event.payload("sequence_len").value = i % 10
+    sequence_field = event.payload("sequence_field")
+    sequence_field.length = event.payload("sequence_len")
+    for j in range(event.payload("sequence_len").value):
+        sequence_field.field(j).value = i + j
 
-       event.payload("sequence_len").value = i % 10
-       sequence_field = event.payload("sequence_field")
-       sequence_field.length = event.payload("sequence_len")
-       for j in range(event.payload("sequence_len").value):
-               sequence_field.field(j).value = i + j
+    enumeration_field = event.payload("enum_field")
+    integer_field = enumeration_field.container
+    enumeration_field.value = i % 10
 
-       enumeration_field = event.payload("enum_field")
-       integer_field = enumeration_field.container
-       enumeration_field.value = i % 10
+    stream.append_event(event)
 
-       stream.append_event(event)
+# Populate custom packet context field before flushing
+packet_context = stream.packet_context
+field = packet_context.field("a_custom_packet_context_field")
+field.value = 42
 
 stream.flush()