Use proper arg parser

This commit is contained in:
Christophe Bedard 2019-06-06 09:29:23 +02:00
parent 63fd1de4ba
commit d55a7f4f3c

View file

@ -1,9 +1,9 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# Entrypoint/script to setup and start an LTTng tracing session # Entrypoint/script to setup and start an LTTng tracing session
# TODO
import sys import sys
import time import time
import argparse
from tracetools_trace.tools.lttng import ( from tracetools_trace.tools.lttng import (
lttng_setup, lttng_setup,
lttng_start, lttng_start,
@ -11,13 +11,20 @@ from tracetools_trace.tools.lttng import (
lttng_destroy, lttng_destroy,
) )
def main(argv=sys.argv): def main():
if len(argv) != 3: parser = argparse.ArgumentParser(description='Setup and launch an LTTng tracing session.')
print("usage: session-name /path") parser.add_argument('--session-name', '-s', dest='session_name',
exit(1) default=f'session-{time.strftime("%Y%m%d%H%M%S")}',
help='the name of the tracing session (default: session-YYYYMMDDHHMMSS)')
parser.add_argument('--path', '-p', dest='path',
default='/tmp',
help='path of the base directory for trace data (default: %(default)s)')
args = parser.parse_args()
session_name = args.session_name
base_path = args.path
path = base_path + '/' + session_name
session_name = argv[1]
path = argv[2] + '/' + session_name
lttng_setup(session_name, path) lttng_setup(session_name, path)
lttng_start(session_name) lttng_start(session_name)
print(f'tracing session started: {path}') print(f'tracing session started: {path}')