cmdrun

gistfile1.py

import subprocess


class ProcessError(IOError):
    pass


def cmdrun(cli,
           debug=False,
           shell=True,
           env=None,
           cwd=None,
           wait=True,
           stdout=subprocess.PIPE,
           stderr=subprocess.PIPE,
           raise_error=True,
           stdin=None,
           *args,
           **kwargs):
    if cwd is None:
        cwd = os.getcwd()
    if env is None:
        env = copy.deepcopy(os.environ)
    kwargs.setdefault('cwd', cwd)
    kwargs.setdefault('stdout', stdout)
    kwargs.setdefault('stderr', stderr)
    kwargs.setdefault('stdin', stdin)
    kwargs.setdefault('env', env)
    out, err = None, None
    if isinstance(cli, six.string_types):
        cli = shlex.split(cli)
    if debug:
        print(' '.join(cli), file=sys.stderr)
    pr = subprocess.Popen(cli, *args, **kwargs)
    if wait:
        out, err = pr.communicate()
        if raise_error and pr.returncode != 0:
            if out:
                print(out, file=sys.stderr)
            if err:
                print(err, file=sys.stderr)
            raise ProcessError(cli)
    return {'out': out, 'err': err, 'pr': pr}