# HG changeset patch
# User Tero Marttila <terom@paivola.fi>
# Date 1394282240 -7200
# Node ID 1053e69664a699e9ef6e1048cf948f0ef72f020a
# Parent  69e1b91cd83f99084923ec58ea93e844ecf502ee
pvl.invoke: setenv

diff -r 69e1b91cd83f -r 1053e69664a6 pvl/invoke.py
--- a/pvl/invoke.py	Fri Dec 27 00:35:10 2013 +0200
+++ b/pvl/invoke.py	Sat Mar 08 14:37:20 2014 +0200
@@ -2,8 +2,9 @@
     Invoke external commands, with python kwargs -> options mangling.
 """
 
+import logging
+import os
 import subprocess
-import logging
 
 log = logging.getLogger('pvl.invoke')
 
@@ -16,7 +17,7 @@
     def __str__ (self) :
         return "{self.cmd} failed ({self.exit}): {self.error}".format(self=self)
 
-def invoke (cmd, args, stdin=None) :
+def invoke (cmd, args, stdin=None, setenv=None) :
     """
         Invoke a command directly.
         
@@ -45,7 +46,14 @@
         io = subprocess.PIPE
         input = None
 
-    p = subprocess.Popen([cmd] + args, stdin=io, stdout=io, stderr=io)
+    if setenv :
+        env = dict(os.environ)
+        env.update(setenv)
+
+    else :
+        env = None
+
+    p = subprocess.Popen([cmd] + list(args), stdin=io, stdout=io, stderr=io, env=env)
 
     # get output
     # returns None if not io