From e1bc6caa509afc1df671a21b94d33906a88930d9 Mon Sep 17 00:00:00 2001
From: "Jason R. Coombs" <jaraco@jaraco.com>
Date: Mon, 16 Feb 2015 09:51:51 -0500
Subject: Unload all pkg_resources modules and not just the main module. Fixes
 #345.

---
 ez_setup.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'ez_setup.py')

diff --git a/ez_setup.py b/ez_setup.py
index 4bffc14c..0734d6c5 100644
--- a/ez_setup.py
+++ b/ez_setup.py
@@ -181,7 +181,12 @@ def use_setuptools(
 
 
 def _unload_pkg_resources():
-    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]
 
 
 def _clean_check(cmd, target):
-- 
cgit v1.2.1