Reduce the doctest lease duration...

[python_utils.git] / config.py

diff --git a/config.py b/config.py
index 0c4016675d7124c3f48e1a9def75794f9b86409d..1a362a381eeaac36b6b8fabafa2db548ee82b1f0 100644 (file)
--- 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:
@@ -234,6 +240,9 @@ class Config:
     def __contains__(self, key: str) -> bool:
         return key in self.config
 
+    def get(self, key: str, default: Any = None) -> Optional[Any]:
+        return self.config.get(key, default)
+
     @staticmethod
     def add_commandline_args(title: str, description: str = "") -> argparse._ArgumentGroup:
         """Create a new context for arguments and return a handle.
@@ -303,7 +312,6 @@ class Config:
     @staticmethod
     def _parse_arg_into_env(arg: str) -> Optional[Tuple[str, str, List[str]]]:
         """Internal helper to parse commandling args into environment vars."""
-
         arg = arg.strip()
         if not arg.startswith('['):
             return None
@@ -314,7 +322,7 @@ class Config:
 
         chunks = arg.split()
         if len(chunks) > 1:
-            var = arg.split()[0]
+            var = chunks[0]
         else:
             var = arg
 
@@ -323,7 +331,7 @@ class Config:
         env = var.upper()
         while env[0] == '-':
             env = env[1:]
-        return arg, env, chunks
+        return var, env, chunks
 
     def _augment_sys_argv_from_environment_variables(self):
         """Internal.  Look at the system environment for variables that match
@@ -364,9 +372,9 @@ class Config:
                                 from string_utils import to_bool
 
                                 if len(chunks) == 1 and to_bool(value):
-                                    sys.argv.append(arg)
+                                    sys.argv.append(var)
                                 elif len(chunks) > 1:
-                                    sys.argv.append(arg)
+                                    sys.argv.append(var)
                                     sys.argv.append(value)
                     arg = ''
 
@@ -572,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: