Tamarin/lib/tamarin.py

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_base_dir():
  return os.path.realpath(os.path.dirname(os.path.abspath(__file__)) + "/..")

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 get_utils_dir():
  return os.path.realpath(os.path.dirname(os.path.abspath(__file__)) + "/../utils")

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 get_datasets_dir():
    return os.path.realpath(os.path.dirname(os.path.abspath(__file__)) + "/../datasets")


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)