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-11-10 17:35:43 -0600 |
commit | 987023ba91d38a58c2fd4269bdfd0bf40648b7de (patch) | |
tree | 5538feb262f6bfa6d3d1f9052226c0a386843cdb /src/lib | |
parent | dc9e04ecd8e7cedb4c1645044828442f264c8c9d (diff) | |
download | bcfg2-987023ba91d38a58c2fd4269bdfd0bf40648b7de.tar.gz bcfg2-987023ba91d38a58c2fd4269bdfd0bf40648b7de.tar.bz2 bcfg2-987023ba91d38a58c2fd4269bdfd0bf40648b7de.zip |
call shutdown on subcommand registries
Diffstat (limited to 'src/lib')
-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() |