diff options
Diffstat (limited to 'support/lib/util.py')
| -rw-r--r-- | support/lib/util.py | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/support/lib/util.py b/support/lib/util.py index 3c5117d..54b92f0 100644 --- a/support/lib/util.py +++ b/support/lib/util.py @@ -27,15 +27,14 @@ def TemporaryDirectory(): shutil.rmtree(dir) -def command(args, remove_env = None): - if remove_env is None: - env = None - else: - env = dict(os.environ) - - for i in remove_env: - if i in env: - del env[i] +def command(args, remove_env = [], set_env = { }): + env = dict(os.environ) + + for i in remove_env: + del env[i] + + for k, v in set_env.items(): + env[k] = v process = subprocess.Popen(args, env = env) process.wait() |