u1234x1234 / tempenviron

Temporary modify your environment variables with context manager.

tempenviron

tempenviron allows you to temporary modify your environment variables with context manager.

Install

pip install tempenviron

Usage

from tempenviron import updated_environ


# Update with keyword arg
assert 'RANDOM_ENVIRONMENT_VARIABLE' not in os.environ

with updated_environ(RANDOM_ENVIRONMENT_VARIABLE='SOME_VALUE'):
    assert os.environ['RANDOM_ENVIRONMENT_VARIABLE'] == 'SOME_VALUE'

assert 'RANDOM_ENVIRONMENT_VARIABLE' not in os.environ


# Or pass a dictionary
with updated_environ({'var1': 'val1', 'var2': 'val2'}):
    assert os.environ['var1'] == 'val1'
    assert os.environ['var2'] == 'val2'


# Or both. In that case keywords have a higher priority
with updated_environ({'var1': 'val1', 'var2': 'val2'}, var2='val3'):
    assert os.environ['var2'] == 'val3'


# To temporary delete an environment variable, set the variable to None
os.environ['var1'] = 'val1'

with updated_environ(var1=None):
    assert 'var1' not in os.environ

assert os.environ['var1'] == 'val1'