Esempio n. 1
0
def set_environment(verbose=False):
    hostname = platform.node()
    env_file = path(__file__).parent.joinpath("environments").joinpath("env_vars.yml")
    if not env_file.exists():
        if verbose:
            print "env_vars file not found."
        return
    else:
        envs = yaml.load(env_file.open())
        resolve_env_vars(envs)
        for k, v in envs.iteritems():
            original_values = os.environ.get(k, "").split(":")
            if v:
                values = v
            else:
                values = []
            combined_values = [v for v in values + original_values if v]
            os.environ[k] = ":".join(combined_values)
            if verbose:
                print "ENV:: ", k, " = ", os.environ[k]
Esempio n. 2
0
    def __init__(self, out_path, exe_path, params, time=10080, 
                priority=1, connection=None, verbose=False, 
                test=False, env='BaseTrial.yml'):
 
        self._connection = None
        if connection:
            self.connection = connection
        self.test = test
        self.verbose = verbose
        self.out_path = out_path
        self.exe_path = exe_path
        self.id_ = None
        self.priority = priority
        self.time = time

        if isinstance(params, dict):
            params = sorted(list(params.iteritems()))
        self.params = sorted(params)
        if verbose:
            print self.params
        sha1 = hashlib.sha1(str((self.exe_path, self.params)))
        self.hash_path = path(sha1.hexdigest())
        env_file = path('./environments') / path(env)
        self.wrap_path = path('./')
       
        if env_file.isfile():
            env = yaml.load(open(env_file))
            self.env = env
            resolve_env_vars(env)
            if self.verbose:
                print 'outpath before sub ', self.out_path
            self.out_path = resolve(env, self.out_path)
            self.wrap_path = resolve(env, Trial.MANAGER)
            self.python_path = resolve(env, Trial.WORK) / 'local/bin/python'
        elif verbose:
            print 'No Enviroment path found'
        if self.verbose:
            print  'exe path, out path, ', self.exe_path, self.out_path