Move print statements to entrypoints and add events list display option

This commit is contained in:
Christophe Bedard 2019-06-06 09:30:48 +02:00
parent 65f50e3981
commit 99b6b0d577
2 changed files with 11 additions and 2 deletions

View file

@ -18,8 +18,6 @@ def lttng_setup(session_name, directory, ros_events=DEFAULT_EVENTS_ROS, kernel_e
""" """
ust_enabled = ros_events is not None and len(ros_events) > 0 ust_enabled = ros_events is not None and len(ros_events) > 0
kernel_enabled = kernel_events is not None and len(kernel_events) > 0 kernel_enabled = kernel_events is not None and len(kernel_events) > 0
print(f'UST tracing {f"enabled ({len(ros_events)} events)" if ust_enabled else "disabled"}')
print(f'kernel tracing {f"enabled ({len(kernel_events)} events)" if kernel_enabled else "disabled"}')
# Domains # Domains
if ust_enabled: if ust_enabled:

View file

@ -27,6 +27,8 @@ def main():
help='the ROS UST events to enable (default: all events) [to disable all UST events, provide this flag without any event name]') help='the ROS UST events to enable (default: all events) [to disable all UST events, provide this flag without any event name]')
parser.add_argument('--kernel', '-k', nargs='*', dest='events_kernel', default=DEFAULT_EVENTS_KERNEL, parser.add_argument('--kernel', '-k', nargs='*', dest='events_kernel', default=DEFAULT_EVENTS_KERNEL,
help='the kernel events to enable (default: all events) [to disable all UST events, provide this flag without any event name]') help='the kernel events to enable (default: all events) [to disable all UST events, provide this flag without any event name]')
parser.add_argument('--list', '-l', dest='list', action='store_true',
help='display lists of enabled events (default: %(default)s)')
args = parser.parse_args() args = parser.parse_args()
session_name = args.session_name session_name = args.session_name
@ -35,6 +37,15 @@ def main():
ros_events = args.events_ust ros_events = args.events_ust
kernel_events = args.events_kernel kernel_events = args.events_kernel
ust_enabled = len(ros_events) > 0
kernel_enabled = len(kernel_events) > 0
print(f'UST tracing {f"enabled ({len(ros_events)} events)" if ust_enabled else "disabled"}')
if args.list and ust_enabled:
print(f'\tevents: {ros_events}')
print(f'kernel tracing {f"enabled ({len(kernel_events)} events)" if kernel_enabled else "disabled"}')
if args.list and kernel_enabled:
print(f'\tevents: {kernel_events}')
lttng_setup(session_name, path, ros_events=ros_events, kernel_events=kernel_events) lttng_setup(session_name, path, ros_events=ros_events, kernel_events=kernel_events)
print(f'writting tracing session to: {path}') print(f'writting tracing session to: {path}')