Commit 813855aa authored by Jason R. Coombs's avatar Jason R. Coombs

Avoid passing None values to os.environ.

parent 475e7c86
...@@ -66,13 +66,16 @@ def patch_env(**replacements): ...@@ -66,13 +66,16 @@ def patch_env(**replacements):
for key in replacements for key in replacements
if key in os.environ if key in os.environ
) )
os.environ.update(replacements)
# remove values that are null # remove values that are null
null_keys = (key for (key, value) in replacements.items() if value is None) remove = (key for (key, value) in replacements.items() if value is None)
list(map(os.environ.pop, (null_keys))) for key in list(remove):
os.environ.pop(key, None)
replacements.pop(key)
os.environ.update(replacements)
yield yield saved
for key in replacements: for key in replacements:
os.environ.pop(key, None) os.environ.pop(key, None)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment