bindings/python/examples/example-api-test.py

   1 #!/usr/bin/env python3
   2 # example_api_test.py
   3 #
   4 # Babeltrace example script based on the Babeltrace API test script
   5 #
   6 # Copyright 2012 EfficiOS Inc.
   7 #
   8 # Author: Danny Serres <danny.serres@efficios.com>
   9 #
  10 # Permission is hereby granted, free of charge, to any person obtaining a copy
  11 # of this software and associated documentation files (the "Software"), to deal
  12 # in the Software without restriction, including without limitation the rights
  13 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  14 # copies of the Software, and to permit persons to whom the Software is
  15 # furnished to do so, subject to the following conditions:
  16 #
  17 # The above copyright notice and this permission notice shall be included in
  18 # all copies or substantial portions of the Software.
  19 #
  20 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  21 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  22 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  23 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  24 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  25 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  26 # SOFTWARE.
  27
  28 # This example uses the babeltrace python module
  29 # to partially test the api.
  30
  31 import sys
  32 from babeltrace import *
  33
  34 # Check for path arg:
  35 if len(sys.argv) < 2:
  36     raise TypeError("Usage: python example-api-test.py path/to/file")
  37
  38 # Create TraceCollection and add trace:
  39 traces = TraceCollection()
  40 trace_handle = traces.add_trace(sys.argv[1], "ctf")
  41 if trace_handle is None:
  42     raise IOError("Error adding trace")
  43
  44 # Listing events
  45 print("--- Event list ---")
  46 for event_declaration in trace_handle.events:
  47     print("event : {}".format(event_declaration.name))
  48     if event_declaration.name == "sched_switch":
  49         for field_declaration in event_declaration.fields:
  50             print(field_declaration)
  51 print("--- Done ---")
  52
  53 for event in traces.events:
  54     print("TS: {}, {} : {}".format(event.timestamp, event.cycles, event.name))
  55
  56     if event.name == "sched_switch":
  57         prev_comm = event["prev_comm"]
  58         if prev_comm is None:
  59             print("ERROR: Missing prev_comm context info")
  60         else:
  61             print("sched_switch prev_comm: {}".format(prev_comm))
  62
  63     if event.name == "exit_syscall":
  64         ret_code = event["ret"]
  65         if ret_code is None:
  66             print("ERROR: Unable to extract ret")
  67         else:
  68             print("exit_syscall ret: {}".format(ret_code))