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):
for key in replacements
if key in os.environ
)
os.environ.update(replacements)
# remove values that are null
null_keys = (key for (key, value) in replacements.items() if value is None)
list(map(os.environ.pop, (null_keys)))
remove = (key for (key, value) in replacements.items() if value is None)
for key in list(remove):
os.environ.pop(key, None)
replacements.pop(key)
os.environ.update(replacements)
yield
yield saved
for key in replacements:
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