B ½4¹[B ã@s:ddlZddlZejdd„ƒZejd dd„ƒZdd„ZdS) éNc csHtj ¡}tj ¡tj |¡z dVWdtj ¡tj |¡XdS)z²Completely replace the environment variables with the specified dict. Use as a context manager:: with temporary_env({'PATH': my_path}): ... N)ÚosÚenvironÚcopyÚclearÚupdate)ZnewenvÚorig_env©rú+lib/python3.7/site-packages/testpath/env.pyÚ temporary_envs     r Tc cs‚dd„}|rtj ¡}n,i}x&| ¡D]\}}tj |d¡||<q&W||ƒz dVWd|rttj ¡tj |¡n||ƒXdS)a™Temporarily modify environment variables. Specify the changes as a dictionary mapping names to new values, using None as the value for names that should be deleted. Example use:: with modified_env({'SHELL': 'bash', 'PYTHONPATH': None}): ... When the context exits, there are two possible ways to restore the environment. If *snapshot* is True, the default, it will reset the whole environment to its state when the context was entered. If *snapshot* is False, it will restore only the specific variables it modified, leaving any changes made to other environment variables in the context. cSs<x6| ¡D]*\}}|dkr*tj |d¡q |tj|<q WdS)N)ÚitemsrrÚpop)ÚchangesÚkÚvrrr Ú update_del(sz modified_env..update_delN)rrrr Úgetrr)r ZsnapshotrZsaved_variablesrrrrr Ú modified_envs   rcstj ¡‰‡fdd„}|S)a…Snapshot the current environment, return a function to restore that. This is intended to produce cleanup functions for tests. For example, using the :class:`unittest.TestCase` API:: def setUp(self): self.addCleanup(testpath.make_env_restorer()) Any changes a test makes to the environment variables will be wiped out before the next test is run. cstj ¡tj ˆ¡dS)N)rrrrr)rrr ÚrestoreOs z"make_env_restorer..restore)rrr)rr)rr Úmake_env_restorerAs  r)T)Ú contextlibrÚcontextmanagerr rrrrrr Ús  *