6264: Support setting environment variables.
authorPeter Amstutz <peter.amstutz@curoverse.com>
Wed, 1 Jul 2015 21:43:28 +0000 (17:43 -0400)
committerPeter Amstutz <peter.amstutz@curoverse.com>
Wed, 1 Jul 2015 21:43:28 +0000 (17:43 -0400)
crunch_scripts/run-command

index 1ff63616ef638db376d49cfeecf8655f04dddbb1..a6c5ef981ce35fb5e52f8bab9c6628c803e45872 100755 (executable)
@@ -331,6 +331,13 @@ try:
         if not args.dry_run:
             stdoutfile = open(stdoutname, "wb")
 
+    if "task.env" in taskp:
+        env = copy.copy(os.environ)
+        for k,v in taskp["task.env"].items():
+            env[k] = subst.do_substitution(taskp, v)
+    else:
+        env = None
+
     logger.info("{}{}{}".format(' | '.join([' '.join(c) for c in cmd]), (" < " + stdinname) if stdinname is not None else "", (" > " + stdoutname) if stdoutname is not None else ""))
 
     if args.dry_run:
@@ -363,7 +370,7 @@ try:
             # this is an intermediate command in the pipeline, so its stdout should go to a pipe
             next_stdout = subprocess.PIPE
 
-        sp = subprocess.Popen(cmd[i], shell=False, stdin=next_stdin, stdout=next_stdout)
+        sp = subprocess.Popen(cmd[i], shell=False, stdin=next_stdin, stdout=next_stdout, env=env)
 
         # Need to close the FDs on our side so that subcommands will get SIGPIPE if the
         # consuming process ends prematurely.