niklas/ros2_tracing
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add completers for event names

Browse source
This commit is contained in:
Christophe Bedard 2019-06-23 18:18:47 +02:00
parent 10210566a6
commit f2bb28215e
1 changed files with 15 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tracetools_trace/tools/args.py
View file

@ -4,6 +4,17 @@ import time
from . import names from . import names
class DefaultArgValueCompleter:
"""Callable returning an arg's default value."""
def __init__(self, arg):
default = arg.default
self.list = default if isinstance(default, list) else [default]
def __call__(self, **kwargs):
return self.list
def parse_args(): def parse_args():
""" """
Parse args for tracing. Parse args for tracing.
@ -22,17 +33,19 @@ def add_arguments(parser):
'--path', '-p', dest='path', '--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( arg = parser.add_argument(
'--ust', '-u', nargs='*', dest='events_ust', default=names.DEFAULT_EVENTS_ROS, '--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( arg.completer = DefaultArgValueCompleter(arg)
arg = parser.add_argument(
'--kernel', '-k', nargs='*', dest='events_kernel', '--kernel', '-k', nargs='*', dest='events_kernel',
default=names.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]')
arg.completer = DefaultArgValueCompleter(arg)
parser.add_argument( parser.add_argument(
'--list', '-l', dest='list', action='store_true', '--list', '-l', dest='list', action='store_true',
help='display lists of enabled events (default: %(default)s)') help='display lists of enabled events (default: %(default)s)')