コード例 #1
0
ファイル: init.py プロジェクト: xiaming9880/pai
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")
コード例 #2
0
ファイル: init.py プロジェクト: xiaming9880/pai
# 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)
コード例 #3
0
# 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)
コード例 #4
0
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)
コード例 #5
0
ファイル: init.py プロジェクト: qzwlinux/pai
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")