niceSOFT/python3-requests: A simple, yet elegant, HTTP library.

tests/utils.py

19 строк391 B

Nate Prewitt

making module docstrings and coding comments consistent

19 июл 2016, 23:51

19 июл 2016, 23:5115a3869

# -*- coding: utf-8 -*-

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)