diff options
| -rw-r--r-- | sos/__init__.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/sos/__init__.py b/sos/__init__.py index 49a753be..10f67b93 100644 --- a/sos/__init__.py +++ b/sos/__init__.py @@ -62,6 +62,14 @@ class SoSComponent(): self.parser = parser self.args = parsed_args self.cmdline = cmdline_args + self.exit_process = False + + try: + import signal + signal.signal(signal.SIGTERM, self.get_exit_handler()) + except Exception: + pass + # update args from component's arg_defaults defintion self._arg_defaults.update(self.arg_defaults) self.opts = self.load_options() @@ -71,6 +79,15 @@ class SoSComponent(): self.tempfile_util = TempFileUtil(self.tmpdir) self._setup_logging() + def get_exit_handler(self): + def exit_handler(signum, frame): + self.exit_process = True + self._exit() + return exit_handler + + def _exit(self, error=0): + raise SystemExit(error) + @classmethod def add_parser_options(cls, parser): """This should be overridden by each subcommand to add its own unique |