diff --git a/kibot/mcpyrate/activate.py b/kibot/mcpyrate/activate.py index f1893b81..1ffe2612 100644 --- a/kibot/mcpyrate/activate.py +++ b/kibot/mcpyrate/activate.py @@ -26,10 +26,23 @@ the `PYTHONDONTWRITEBYTECODE` environment variable, and the attribute ''' from importlib.machinery import SourceFileLoader, FileFinder - from .importer import source_to_xcode, path_xstats, invalidate_xcaches -SourceFileLoader.source_to_code = source_to_xcode -# we could replace SourceFileLoader.set_data with a no-op to force-disable pyc caching. -SourceFileLoader.path_stats = path_xstats -FileFinder.invalidate_caches = invalidate_xcaches + +def activate(): + SourceFileLoader.source_to_code = source_to_xcode + # we could replace SourceFileLoader.set_data with a no-op to force-disable pyc caching. + SourceFileLoader.path_stats = path_xstats + FileFinder.invalidate_caches = invalidate_xcaches + + +def de_activate(): + SourceFileLoader.source_to_code = old_source_to_code + SourceFileLoader.path_stats = old_path_stats + FileFinder.invalidate_caches = old_invalidate_caches + + +old_source_to_code = SourceFileLoader.source_to_code +old_path_stats = SourceFileLoader.path_stats +old_invalidate_caches = FileFinder.invalidate_caches +activate()