From 87396b2710cefe67205c7eff4933e6e810276864 Mon Sep 17 00:00:00 2001 From: Christophe Bedard Date: Thu, 27 Jun 2019 10:35:46 +0200 Subject: [PATCH] Make --pickle-path arg optional --- .../tracetools_analysis/convert.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/tracetools_analysis/tracetools_analysis/convert.py b/tracetools_analysis/tracetools_analysis/convert.py index 576db4c..2268101 100644 --- a/tracetools_analysis/tracetools_analysis/convert.py +++ b/tracetools_analysis/tracetools_analysis/convert.py @@ -16,16 +16,25 @@ """Entrypoint/script to convert CTF trace data to a pickle file.""" import argparse +import os import time from tracetools_analysis.conversion import ctf def parse_args(): - parser = argparse.ArgumentParser(description='Convert CTF trace data to a pickle file.') - parser.add_argument('trace_directory', help='the path to the main CTF trace directory') - parser.add_argument('pickle_path', help='the path to the target pickle file to generate') - return parser.parse_args() + parser = argparse.ArgumentParser( + description='Convert CTF trace data to a pickle file.') + parser.add_argument( + 'trace_directory', help='the path to the main CTF trace directory') + parser.add_argument( + '--pickle-path', '-p', + help='the path to the target pickle file to generate (default: $trace_directory/pickle)') + args = parser.parse_args() + if args.pickle_path is None: + args.pickle_path = os.path.join(args.trace_directory, 'pickle') + print(args.pickle_path) + return args def main():