View: tests/utils.py

← requests / tests/utils.py

import contextlib
import os


@contextlib.contextmanager
def override_environ(**kwargs):
    save_env = dict(os.environ)
    for key, value in kwargs.items():
        if value is None:
            del os.environ[key]
        else:
            os.environ[key] = value
    try:
        yield
    finally:
        os.environ.clear()
        os.environ.update(save_env)


1	import contextlib
2	import os
3
4
5	@contextlib.contextmanager
6	def override_environ(**kwargs):
7	save_env = dict(os.environ)
8	for key, value in kwargs.items():
9	if value is None:
10	del os.environ[key]
11	else:
12	os.environ[key] = value
13	try:
14	yield
15	finally:
16	os.environ.clear()
17	os.environ.update(save_env)
18