def main(): LOGGER.info("Preparing storage runtime plugin commands") [parameters, pre_script, _] = plugin_init() try: command_generator = StorageCommandGenerator() except Exception: #pylint: disable=broad-except LOGGER.exception("Failed to generate storage commands") sys.exit(1) pre_script_commands = command_generator.generate_plugin_commands( parameters) inject_commands(pre_script_commands, pre_script) LOGGER.info("Storage runtime plugin perpared")
# The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. # # THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING # BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, # DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. from __future__ import print_function import os import sys sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")) import collections import logging import argparse import yaml from plugin_utils import plugin_init, inject_commands logger = logging.getLogger(__name__) if __name__ == "__main__": [parameters, pre_script, post_script] = plugin_init() if parameters is not None: if "preCommands" in parameters: inject_commands(parameters["preCommands"], pre_script) if "postCommands" in parameters: inject_commands(parameters["postCommands"], post_script)
# the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and # to permit persons to whom the Software is furnished to do so, subject to the following conditions: # The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. # # THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING # BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, # DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. from __future__ import print_function import os import sys sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")) import logging from plugin_utils import plugin_init, inject_commands logger = logging.getLogger(__name__) if __name__ == "__main__": [parameters, pre_script, post_script] = plugin_init() commands = [] if parameters is not None: logdir = ",".join(["{}:{}".format(k, v) for k, v in parameters["logdir"].items()]) commands.append("tensorboard --logdir={} --port={} &\n".format(logdir, parameters["port"])) inject_commands(commands, pre_script)
import os import sys sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")) import collections import logging import argparse import yaml from plugin_utils import plugin_init, inject_commands logger = logging.getLogger(__name__) if __name__ == "__main__": [parameters, pre_script, post_script] = plugin_init() cmdParams = [] if parameters is not None: if "jobssh" in parameters: cmdParams.append(str(parameters["jobssh"]).lower()) else: cmdParams.append("false") if "userssh" in parameters: if "type" in parameters["userssh"] and "value" in parameters["userssh"]: cmdParams.append(str(parameters["userssh"]["type"])) cmdParams.append("\'{}\'".format(parameters["userssh"]["value"])) # write call to real executable script command = "{}/sshd.sh {}\n".format(os.path.dirname(os.path.abspath(__file__)), " ".join(cmdParams)) inject_commands([command], pre_script)
def main(): LOGGER.info("Preparing storage runtime plugin commands") [parameters, pre_script, _] = plugin_init() pre_script_commands = generate_plugin_commands(parameters) inject_commands(pre_script_commands, pre_script) LOGGER.info("Storage runtime plugin perpared")