Only using ultan on 3.4+

Author: Austin Bingham

setup.py

@@ -5,11 +5,13 @@
 
 install_requires = [
     'rope',
-    'ultan'
 ]
 
 if sys.version_info < (3, 4):
     install_requires.append('enum34')
+else:
+    # We only enable ultan stuff in 3.4+
+    install_requires.append('ultan')
 
 
 def local_file(*name):

traad/rope/auto_import.py

@@ -1,12 +1,39 @@
+from logging import getLogger
+
 from rope.base.worder import get_name_at
 from rope.contrib.autoimport import AutoImport
-from ultan.name_index import NameIndex
+
+log = getLogger()
+
+
+# We use different name-fetching backends depending on whether ultan is
+# available or not.
+try:
+    from ultan.name_index import NameIndex
+
+    class NameGetter:
+        def __init__(self):
+            self.name_index = NameIndex()
+
+        def get_names(self, name):
+            return self.name_index.get_names(name)
+
+except ImportError:
+    log.info('Ultan not available. Falling back to rope auto-import.')
+
+    # We only really come down this path if we're on python 2.7, so we inherit
+    # from object to make super work later. This is a little shady, but that's
+    # how I roll.
+    class NameGetter(object):
+        def get_names(self, name):
+            return self.auto_import.import_assist(name)
 
 
-class AutoImportMixin:
+class AutoImportMixin(NameGetter):
     def __init__(self):
+        super(AutoImportMixin, self).__init__()
         self._auto_import = None
-        self.name_index = NameIndex()
+        self.name_getter = NameGetter()
 
     @property
     def auto_import(self):
@@ -20,9 +47,8 @@ def get_imports(self, path, offset):
             print(r.path)
         res = self.get_resource(path)
         name = get_name_at(res, offset)
-        imports = tuple(self.name_index.get_names(name))
+        imports = tuple(self.get_names(name))
         location = self.auto_import.find_insertion_line(res.read())
-        print(res, name, imports, location)
         return {
             'imports': imports,
             'location': location