Plugin to run bash script

starcluster/plugins/runbash.py

@@ -0,0 +1,56 @@
+# Copyright 2009-2014 Justin Riley
+#
+# This file is part of StarCluster.
+#
+# StarCluster is free software: you can redistribute it and/or modify it under
+# the terms of the GNU Lesser General Public License as published by the Free
+# Software Foundation, either version 3 of the License, or (at your option) any
+# later version.
+#
+# StarCluster is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
+# details.
+#
+# You should have received a copy of the GNU Lesser General Public License
+# along with StarCluster. If not, see <http://www.gnu.org/licenses/>.
+
+"""
+"""
+from starcluster.clustersetup import DefaultClusterSetup
+from starcluster.logger import log
+from starcluster.utils import print_timing
+
+
+class BashRunner(DefaultClusterSetup):
+    """Bash Runner"""
+
+    def __init__(self, bash_file=None):
+        super(BashRunner, self).__init__()
+        self.bash_file = bash_file
+
+    @print_timing("BashRunner")
+    def setup_swap(self, nodes):
+        if not self.bash_file:
+            log.info("No bash file specified!")
+            return
+
+        log.info("Running bash file: %s" % self.bash_file)
+        with open(self.bash_file, 'r') as fp:
+            commands = fp.readlines()
+
+        for command in commands:
+            log.info("$ " + command)
+        cmd = "\n".join(commands)
+        for node in nodes:
+            self.pool.simple_job(node.ssh.execute, (cmd,), jobid=node.alias)
+        self.pool.wait(len(nodes))
+
+    def run(self, nodes, master, user, user_shell, volumes):
+        self.setup_swap(nodes)
+
+    def on_add_node(self, node, nodes, master, user, user_shell, volumes):
+        self.setup_swap([node])
+
+    def on_remove_node(self, node, nodes, master, user, user_shell, volumes):
+        raise NotImplementedError("on_remove_node method not implemented")