[modutils] new cleanup_sys_modules function (backported from yams)

2 files changed, 13 insertions, 0 deletions

diff --git a/ChangeLog b/ChangeLog
index e9d4fed..08cd6d0 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -14,6 +14,9 @@ ChangeLog for logilab.common
 
     * sphinxutils: simplified API
 
+    * modutils: new cleanup_sys_modules function that removes modules under a list
+      of directories from sys.modules
+
 
 2009-07-17  --  0.42.0
     * pyro_ext: new module for pyro utilities
diff --git a/modutils.py b/modutils.py
index 5755a4f..837d2b0 100644
--- a/modutils.py
+++ b/modutils.py
@@ -444,6 +444,16 @@ def get_source_file(filename, include_no_ext=False):
     raise NoSourceFile(filename)
 
 
+def cleanup_sys_modules(directories):
+    """remove submodules of `directories` from `sys.modules`"""
+    for modname, module in sys.modules.items():
+        modfile = getattr(module, '__file__', None)
+        if modfile:
+            for directory in directories:
+                if modfile.startswith(directory):
+                    del sys.modules[modname]
+                    break
+
 
 def is_python_source(filename):
     """