diff options
Diffstat (limited to 'modutils.py')
-rw-r--r-- | modutils.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modutils.py b/modutils.py index 2756841..d0cccf3 100644 --- a/modutils.py +++ b/modutils.py @@ -455,13 +455,16 @@ def get_source_file(filename, include_no_ext=False): def cleanup_sys_modules(directories): """remove submodules of `directories` from `sys.modules`""" + cleaned = [] for modname, module in sys.modules.items(): modfile = getattr(module, '__file__', None) if modfile: for directory in directories: if modfile.startswith(directory): + cleaned.append(modname) del sys.modules[modname] break + return cleaned def is_python_source(filename): |