diff options
author | Chris St. Pierre <chris.a.st.pierre@gmail.com> | 2014-10-22 08:07:28 -0500 |
---|---|---|
committer | Chris St. Pierre <chris.a.st.pierre@gmail.com> | 2014-10-22 13:11:55 -0500 |
commit | 1a54f2a549b86be5a976d2a43a7985c1265d916a (patch) | |
tree | a6be9a1cf46845cdf691e89f8bf6e509170dbec3 | |
parent | a3998969ec25acf20d5e42a1166e001288ca7b0e (diff) | |
download | bcfg2-1a54f2a549b86be5a976d2a43a7985c1265d916a.tar.gz bcfg2-1a54f2a549b86be5a976d2a43a7985c1265d916a.tar.bz2 bcfg2-1a54f2a549b86be5a976d2a43a7985c1265d916a.zip |
call shutdown on subcommand registries
-rw-r--r-- | src/lib/Bcfg2/Server/Admin.py | 7 | ||||
-rw-r--r-- | src/lib/Bcfg2/Server/Info.py | 10 |
2 files changed, 11 insertions, 6 deletions
diff --git a/src/lib/Bcfg2/Server/Admin.py b/src/lib/Bcfg2/Server/Admin.py index b6dded175..ef7741880 100644 --- a/src/lib/Bcfg2/Server/Admin.py +++ b/src/lib/Bcfg2/Server/Admin.py @@ -1197,5 +1197,8 @@ class CLI(Bcfg2.Options.CommandRegistry): def run(self): """ Run bcfg2-admin """ - self.commands[Bcfg2.Options.setup.subcommand].setup() - return self.runcommand() + try: + self.commands[Bcfg2.Options.setup.subcommand].setup() + return self.runcommand() + finally: + self.shutdown() diff --git a/src/lib/Bcfg2/Server/Info.py b/src/lib/Bcfg2/Server/Info.py index 0baa29a9f..da2312cc8 100644 --- a/src/lib/Bcfg2/Server/Info.py +++ b/src/lib/Bcfg2/Server/Info.py @@ -837,7 +837,6 @@ class InfoCore(cmd.Cmd, Bcfg2.Server.Core.Core): pass def shutdown(self): - Bcfg2.Options.CommandRegistry.shutdown(self) Bcfg2.Server.Core.Core.shutdown(self) @@ -868,6 +867,9 @@ class CLI(Bcfg2.Options.CommandRegistry): def run(self): """ Run bcfg2-info """ - if Bcfg2.Options.setup.subcommand != 'help': - self.core.run() - return self.runcommand() + try: + if Bcfg2.Options.setup.subcommand != 'help': + self.core.run() + return self.runcommand() + finally: + self.shutdown() |