Always unload pkg_resources in bundle context manager

zrzka · Jan 11, 2018 · f89eba2 · f89eba2
1 parent 7a07ff3
commit f89eba2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -8,6 +8,9 @@
 * `blackmamba.uikit.keyboard`
     * Camel case enum constant deprecated, will be removed in 2.0.0
     * Use UPPER_CASED versions, camel case still exists as aliases
+* `with bundle()` always unloads `pkg_resources.*` causing issues
+    * Launch Pythonista, run Black Mamba Analyze - works vs
+    * Launch Pythonista, run Pythonista unit tests, run Black Mamba Analyze - fails, because of `pkg_resources`
 
 ## 1.4.3 (2017-12-30)
 

diff --git a/blackmamba/bundle.py b/blackmamba/bundle.py
@@ -11,6 +11,10 @@
     'refactoring': ['rope', 'future', 'libfuturize', 'libpasteurize']
 }
 
+_ALWAYS_UNLOAD = [
+    'pkg_resources'
+]
+
 _BUNDLED_MODULES_PATH = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'lib'))
 
 
@@ -63,6 +67,9 @@ def unload(name):
 
 @contextmanager
 def bundle(name, unload_bundle=False):
+    for module in _ALWAYS_UNLOAD:
+        _unload_modules(module)
+
     load(name)
     yield name
     if unload_bundle: