From 45f088fa2b9a22eb8f61b9fefa22a853911b2afb Mon Sep 17 00:00:00 2001 From: Benjamin Renard Date: Wed, 26 May 2021 11:56:02 +0200 Subject: [PATCH] scripts.helpers: improve get_opts_parser() --- mylib/scripts/helpers.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/mylib/scripts/helpers.py b/mylib/scripts/helpers.py index 9514311..bed6996 100644 --- a/mylib/scripts/helpers.py +++ b/mylib/scripts/helpers.py @@ -30,9 +30,9 @@ def init_logging(options, name, report=None): logging.basicConfig(level=loglevel, format=logformat, handlers=handlers) -def get_opts_parser(just_try=False, progress=False): +def get_opts_parser(desc=None, just_try=False, just_one=False, progress=False): """ Retrieve options parser """ - parser = argparse.ArgumentParser() + parser = argparse.ArgumentParser(description=desc) parser.add_argument( '-v', '--verbose', @@ -71,6 +71,14 @@ def get_opts_parser(just_try=False, progress=False): help="Enable just-try mode" ) + if just_one: + parser.add_argument( + '-J', '--just-one', + action="store_true", + dest="just_one", + help="Enable just-one mode" + ) + if progress: parser.add_argument( '-p', '--progress',