71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
import os, glob, subprocess, configparser, codecs, sys
|
|
|
|
def run_profile_hooks(profile, step, **kwargs):
|
|
hooks_dir = get_hooks_dir()
|
|
step_hooks = profile[step]["hooks"]
|
|
if not step_hooks:
|
|
return
|
|
for hook_name in step_hooks.split(","):
|
|
trimmed_hook_name = hook_name.strip(' \t\n\r')
|
|
if not trimmed_hook_name:
|
|
continue
|
|
hook_path = os.path.join(hooks_dir, trimmed_hook_name)
|
|
run([hook_path], **kwargs)
|
|
|
|
def get_hooks_dir():
|
|
return os.path.realpath(os.path.dirname(os.path.abspath(__file__)) + "/../hooks")
|
|
|
|
def get_lib_dir():
|
|
return os.path.realpath(os.path.dirname(os.path.abspath(__file__)) + "/../lib")
|
|
|
|
def load_profile(profile_name, debug=False):
|
|
profile_filename = profile_name+".conf"
|
|
for profile_file in get_available_profiles():
|
|
if profile_filename == os.path.basename(profile_file):
|
|
config = configparser.ConfigParser()
|
|
with codecs.open(profile_file, encoding = 'utf-8', mode = 'r') as handle:
|
|
config.read_file(handle)
|
|
return config
|
|
return None
|
|
|
|
def get_profiles_dir():
|
|
return os.path.realpath(os.path.dirname(os.path.abspath(__file__)) + "/../profiles")
|
|
|
|
def get_available_profiles():
|
|
return glob.glob(get_profiles_dir() + '/*.conf')
|
|
|
|
def get_available_profile_names():
|
|
profile_files = get_available_profiles()
|
|
return [os.path.splitext(os.path.basename(f))[0] for f in profile_files]
|
|
|
|
def get_workspace_dir():
|
|
"""Return the absolute path to the tamarin workspace ($HOME/.tamarin)"""
|
|
home = os.environ["HOME"]
|
|
return os.path.join(os.sep, home, '.tamarin')
|
|
|
|
def get_workspace_subdir(subdir):
|
|
"""Return the absolute path to a subdirectory in tamarin workspace"""
|
|
dir_path = os.path.join(os.sep, get_workspace_dir(), subdir)
|
|
os.makedirs(dir_path, exist_ok=True)
|
|
return dir_path
|
|
|
|
def run(cmd, captureOutput=False, pty=False, debug=False, **kwargs):
|
|
"""Execute an arbitrary command on the system"""
|
|
if debug:
|
|
print(" ".join(cmd) if isinstance(cmd, list) else cmd)
|
|
stdin=subprocess.PIPE
|
|
kwargs['shell'] = False if isinstance(cmd, list) else True
|
|
if pty:
|
|
kwargs['stdin'] = sys.stdin
|
|
if captureOutput:
|
|
return subprocess.check_output(cmd, **kwargs)
|
|
else:
|
|
return subprocess.check_call(cmd, **kwargs)
|
|
|
|
def run_docker(args, captureOutput=False, **kwargs):
|
|
if isinstance(args, list):
|
|
cmd = ["docker"] + args
|
|
else:
|
|
cmd = "docker " + args
|
|
return run(cmd, captureOutput=captureOutput, **kwargs)
|