bumblebee-status/util/cli.py

import os
import shlex
import subprocess
import logging


def execute(cmd, wait=True, ignore_errors=False, include_stderr=False, env=None):
    args = shlex.split(cmd)
    logging.debug(cmd)
    try:
        proc = subprocess.Popen(
            args,
            stdout=subprocess.PIPE,
            stderr=subprocess.STDOUT if include_stderr else subprocess.PIPE,
            env=env,
        )
    except FileNotFoundError as e:
        raise RuntimeError("{} not found".format(cmd))

    if wait:
        out, _ = proc.communicate()
        if proc.returncode != 0:
            err = "{} exited with code {}".format(cmd, proc.returncode)
            if ignore_errors:
                return err
            raise RuntimeError(err)
        return out.decode("utf-8")
    return ""


# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4