54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
"""System related functions."""
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from autonomic import logger
|
|
from autonomic.settings import PASSWORD_STORE_PATH
|
|
|
|
log = logger.get_logger(__name__)
|
|
|
|
|
|
def run_command(command, cwd=None, output=False, extra_env=None):
|
|
"""Run a command."""
|
|
from autonomic.config import get_from_config
|
|
|
|
username = get_from_config("username")
|
|
|
|
env = os.environ.copy()
|
|
env.update({"PASSWORD_STORE_DIR": PASSWORD_STORE_PATH})
|
|
env.update({"REMOTE_USER": username})
|
|
env.update({"ANSIBLE_USER": username})
|
|
|
|
if extra_env:
|
|
for key, val in extra_env.items():
|
|
env.update({key: val})
|
|
|
|
try:
|
|
if cwd:
|
|
os.chdir(cwd)
|
|
|
|
log.info(
|
|
"Running {}{}".format(
|
|
" ".join(command), " in {}".format(cwd) if cwd else ""
|
|
)
|
|
)
|
|
|
|
if output:
|
|
subprocess.call(command, env=env)
|
|
else:
|
|
subprocess.check_output(command, env=env, stderr=subprocess.STDOUT)
|
|
except subprocess.CalledProcessError as exception:
|
|
msg = "{} failed! Saw {}".format(" ".join(command), str(exception))
|
|
exit_with_msg(msg)
|
|
|
|
|
|
def exit(code=1):
|
|
sys.exit(code)
|
|
|
|
|
|
def exit_with_msg(msg, code=1):
|
|
log.critical(msg)
|
|
exit(code)
|