mccabe fixes

blackmamba/__init__.py

@@ -177,9 +177,24 @@ def _check_compatibility_and_updates():
     blackmamba.update.check()
 
 
+def setup_lib_path():
+    """
+    Add blackmamba.lib to sys.path if it's not there.
+    """
+    import sys
+    import os
+
+    lib = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'lib'))
+    try:
+        sys.path.index(lib)
+    except ValueError:
+        sys.path.insert(0, lib)
+
+
 @system.Pythonista()
 @system.catch_exceptions
 def _main(config=None):
+    setup_lib_path()
     # It's here because Sphinx doesn't show documentation for decorated
     # functions
     from blackmamba.config import update_config_with_dict, get_config_value
@@ -222,19 +237,5 @@ def main(config=None):
     _main(config)
 
 
-def setup_lib_path():
-    """
-    Add blackmamba.lib to sys.path if it's not there.
-    """
-    import sys
-    import os
-
-    lib = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'lib'))
-    try:
-        sys.path.index(lib)
-    except ValueError:
-        sys.path.insert(0, lib)
-
-
 if __name__ == '__main__':
     main()

blackmamba/lib/setuptools.py

@@ -0,0 +1 @@
+__version__ = '0.0.1-blackmamba'

blackmamba/script/analyze.py

@@ -8,6 +8,9 @@
 This script is configurable, see :ref:`configuration`.
 """
 
+import blackmamba
+blackmamba.setup_lib_path()
+
 import io
 import pep8
 import re
@@ -212,20 +215,17 @@ def get_style(message):
 
 
 def _flake8_annotations(path, options):
-    import blackmamba
     import os
 
-    blackmamba.setup_lib_path()
-
-    from flake8.main import application
-
     _tmp = os.environ.get('TMPDIR', os.environ.get('TMP'))
     _output_file = os.path.join(_tmp, 'blackmamba.flake8.txt')
 
     annotations = []
 
     for o in options:
         try:
+            from flake8.main import application
+
             if os.path.exists(_output_file):
                 os.remove(_output_file)