Source code for idmtools_slurm_utils.bash

Handles interaction with bash command.

Copyright 2021, Bill & Melinda Gates Foundation. All rights reserved.
import subprocess
from logging import DEBUG, getLogger
from pathlib import Path
from typing import Dict

ERROR_NO_WORKING_DIRECTORY = "No working directory provided for command"

logger = getLogger()

[docs]def command_bash(info: Dict) -> Dict: """ Process command request for sbatch commands to be executed. Args: info: Info command Returns: Result dict """ if 'working_directory' in info: wd = Path(info['working_directory']) if not wd.exists(): output = "FAILED: No Directory name %s" % info['working_directory'] return_code = -1 else: output, return_code = run_bash(wd) result = dict( status="success" if return_code == 0 else "error", return_code=return_code, output=output ) else: result = dict( status="error", return_code=-1, output=ERROR_NO_WORKING_DIRECTORY ) return result
[docs]def run_bash(working_directory: Path): """ Just a bash script. Args: working_directory: Working directory """ sbp = working_directory.joinpath("") if not sbp.exists(): return f"FAILED: No Directory name {sbp}" if logger.isEnabledFor(DEBUG): logger.debug(f"Running 'bash' in {working_directory}") result =['bash', ''], stdout=subprocess.PIPE, cwd=str(working_directory)) return '', result.returncode