Unload all pkg_resources modules and not just the main module. Fixes #453.

1 files changed, 6 insertions, 1 deletions

diff --git a/ez_setup.py b/ez_setup.py
index 50e0dfc6..bbfe0544 100644
--- a/ez_setup.py
+++ b/ez_setup.py
@@ -125,7 +125,12 @@ def _do_download(version, download_base, to_dir, download_delay):
     # Remove previously-imported pkg_resources if present (see
     # https://bitbucket.org/pypa/setuptools/pull-request/7/ for details).
     if 'pkg_resources' in sys.modules:
-        del sys.modules['pkg_resources']
+        del_modules = [
+            name for name in sys.modules 
+            if name.startswith('pkg_resources')
+        ]
+        for mod_name in del_modules:
+            del sys.modules[mod_name]
 
     import setuptools
     setuptools.bootstrap_install_from = egg