Simplify imports

This commit is contained in:
Christophe Bedard 2019-06-07 10:42:40 +02:00
parent 706924dfcf
commit b541f33760

View file

@ -4,16 +4,8 @@
import argparse import argparse
import time import time
from tracetools_trace.tools.lttng import ( from tracetools_trace.tools import lttng
lttng_destroy, from tracetools_trace.tools import names
lttng_setup,
lttng_start,
lttng_stop,
)
from tracetools_trace.tools.names import (
DEFAULT_EVENTS_KERNEL,
DEFAULT_EVENTS_ROS,
)
def main(): def main():
@ -24,12 +16,12 @@ def main():
parser.add_argument('--path', '-p', dest='path', parser.add_argument('--path', '-p', dest='path',
default='/tmp', default='/tmp',
help='path of the base directory for trace data (default: %(default)s)') help='path of the base directory for trace data (default: %(default)s)')
parser.add_argument('--ust', '-u', nargs='*', dest='events_ust', default=DEFAULT_EVENTS_ROS, parser.add_argument('--ust', '-u', nargs='*', dest='events_ust', default=names.DEFAULT_EVENTS_ROS,
help='the ROS UST events to enable (default: all events) ' help='the ROS UST events to enable (default: all events) '
'[to disable all UST events, ' '[to disable all UST events, '
'provide this flag without any event name]') 'provide this flag without any event name]')
parser.add_argument('--kernel', '-k', nargs='*', dest='events_kernel', parser.add_argument('--kernel', '-k', nargs='*', dest='events_kernel',
default=DEFAULT_EVENTS_KERNEL, default=names.DEFAULT_EVENTS_KERNEL,
help='the kernel events to enable (default: all events) ' help='the kernel events to enable (default: all events) '
'[to disable all UST events, ' '[to disable all UST events, '
'provide this flag without any event name]') 'provide this flag without any event name]')
@ -58,13 +50,13 @@ def main():
else: else:
print('kernel tracing disabled') print('kernel tracing disabled')
lttng_setup(session_name, full_path, ros_events=ros_events, kernel_events=kernel_events) lttng.lttng_setup(session_name, full_path, ros_events=ros_events, kernel_events=kernel_events)
print(f'writting tracing session to: {full_path}') print(f'writting tracing session to: {full_path}')
input('press enter to start...') input('press enter to start...')
lttng_start(session_name) lttng.lttng_start(session_name)
input('press enter to stop...') input('press enter to stop...')
print('stopping & destroying tracing session') print('stopping & destroying tracing session')
lttng_stop(session_name) lttng.lttng_stop(session_name)
lttng_destroy(session_name) lttng.lttng_destroy(session_name)