From 91df4fa006f7f4fba8e38a97941c5a4ad9b83706 Mon Sep 17 00:00:00 2001 From: Scott Gasch Date: Thu, 18 Aug 2022 08:54:59 -0700 Subject: [PATCH] Add an option to exit immediately upon setting zookeeper. Change the verbosity of a warning. --- config.py | 11 +++++++++++ logging_utils.py | 2 +- 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/config.py b/config.py index c883a3b..1a362a3 100644 --- a/config.py +++ b/config.py @@ -188,6 +188,12 @@ GROUP.add_argument( action='store_true', help='If present, config will raise an exception if it doesn\'t recognize an argument. The default behavior is to ignore unknown arguments so as to allow interoperability with programs that want to use their own argparse calls to parse their own, separate commandline args.', ) +GROUP.add_argument( + '--config_exit_after_parse', + default=False, + action='store_true', + help='If present, halt the program after parsing config. Useful, for example, to write a --config_savefile and then terminate.', +) class Config: @@ -574,6 +580,11 @@ class Config: self.dump_config() self.config_parse_called = True + if config['config_exit_after_parse']: + print("Exiting because of --config_exit_after_parse.") + if self.zk: + self.zk.stop() + sys.exit(0) return self.config def has_been_parsed(self) -> bool: diff --git a/logging_utils.py b/logging_utils.py index 39453b4..f7e3be6 100644 --- a/logging_utils.py +++ b/logging_utils.py @@ -427,7 +427,7 @@ def log_about_logging( level_name = logging._levelToName.get(default_logging_level, str(default_logging_level)) logger.debug('Initialized global logging; default logging level is %s.', level_name) if config.config['logging_clear_preexisting_handlers'] and preexisting_handlers_count > 0: - logger.warning( + logger.debug( 'Logging cleared %d global handlers (--logging_clear_preexisting_handlers)', preexisting_handlers_count, ) -- 2.45.2