fixed issue with import_module

toumorokoshi · Jan 28, 2015 · 7b8c44f · 7b8c44f
1 parent 4335eba
commit 7b8c44f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@
 ]
 
 setup(name='uranium',
-      version='0.0.39',
+      version='0.0.40',
       description='a build system for python',
       long_description='a build system for python',
       author='Yusuke Tsutsumi',

diff --git a/uranium/classloader.py b/uranium/classloader.py
@@ -31,8 +31,11 @@ def get_class_from_spec(self, class_spec):
         if ':' in class_spec:
             egg_name, module_path = class_spec.split(':')
 
-        if not import_module(module_path) and egg_name:
-            self._install_egg(egg_name)
+        try:
+            import_module(module_path)
+        except ImportError:
+            if egg_name:
+                self._install_egg(egg_name)
 
         return self.get_class(module_path)