19 lines
364 B
Python
19 lines
364 B
Python
|
import contextlib
|
||
|
|
|
||
|
|
from .py39 import import_helper
|
||
|
|
|
||
|
|
|
||
|
|
@contextlib.contextmanager
|
||
|
|
def isolated_modules():
|
||
|
|
"""
|
||
|
|
Save modules on entry and cleanup on exit.
|
||
|
|
"""
|
||
|
|
(saved,) = import_helper.modules_setup()
|
||
|
|
try:
|
||
|
|
yield
|
||
|
|
finally:
|
||
|
|
import_helper.modules_cleanup(saved)
|
||
|
|
|
||
|
|
|
||
|
|
vars(import_helper).setdefault('isolated_modules', isolated_modules)
|