diff options
Diffstat (limited to 'src/lib/Client/Tools/Action.py')
-rw-r--r-- | src/lib/Client/Tools/Action.py | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/src/lib/Client/Tools/Action.py b/src/lib/Client/Tools/Action.py new file mode 100644 index 000000000..ddd15a264 --- /dev/null +++ b/src/lib/Client/Tools/Action.py @@ -0,0 +1,52 @@ +'''Action driver''' +__revision__ = '$Revision$' + +import Bcfg2.Client.Tools + +# <Action timing='pre|post|both' name='name' command='cmd text' when='always|modified' +# status='ignore|check'/> +# <PostInstall name='foo'/> +# => <Action timing='post' when='modified' name='n' command='foo' status='ignore'/> + +class Action(Bcfg2.Client.Tools.Tool): + '''Implement Actions''' + __name__ = 'Action' + __handles__ = [('PostInstall', None), ('Action', None)] + __req__ = {'PostInstall': ['name'], + 'Action':['name', 'timing', 'when', 'command', 'status']} + + def RunAction(self, entry): + '''This method handles command execution and status return''' + self.logger.debug("Running Action %s" % (entry.get('name'))) + rc = self.cmd.run(entry.get('command'))[0] + self.logger.debug("Action: %s got rc %s" % (entry.get('command'), rc)) + if entry.get('status', 'check') == 'ignore': + return True + else: + return rc == 0 + + def VerifyAction(self, dummy, _): + '''Actions always verify true''' + return True + + def InstallAction(self, entry): + '''Run actions as pre-checks for bundle installation''' + if entry.get('timing') != 'post': + self.states[entry] = self.RunAction(entry) + return self.states[entry] + return True + + def BundleUpdated(self, bundle): + '''Run postinstalls when bundles have been updated''' + for postinst in bundle.findall("PostInstall"): + self.cmd.run(postinst.get('name')) + for action in bundle.findall("Action"): + if action.get('timing') in ['post', 'both']: + self.states[action] = self.RunAction(action) + + def BundleNotUpdated(self, bundle): + '''Run Actions when bundles have not been updated''' + for action in bundle.findall("Action"): + if action.get('timing') in ['post', 'both'] and \ + action.get('when') != 'modified': + self.states[action] = self.RunAction(action) |