Remove duplicated code for trace command

This commit is contained in:
Christophe Bedard 2019-12-08 14:01:57 -08:00
parent 9c4e564bb4
commit 3d3cbe9ec3
3 changed files with 58 additions and 75 deletions

View file

@ -15,18 +15,26 @@
"""Module for trace command extension implementation."""
from ros2cli.command import CommandExtension
from ros2trace.api import add_trace_arguments
from ros2trace.api import fini
from ros2trace.api import init
from tracetools_trace.tools import args
from tracetools_trace.trace import init
from tracetools_trace.trace import fini
class TraceCommand(CommandExtension):
"""Trace ROS nodes to get information on their execution."""
def add_arguments(self, parser, cli_name):
add_trace_arguments(parser)
args.add_arguments(parser)
def main(self, *, parser, args):
init(args)
fini(args)
init(
args.session_name,
args.path,
args.events_ust,
args.events_kernel,
args.list,
)
fini(
args.session_name,
)
return 0