k now... emacs>vim

.emacs.d/elpa/elpy-20171206.847/elpy/impmagic.py

@@ -0,0 +1,118 @@
+"""Glue for the "importmagic" library.
+
+"""
+
+import os
+import sys
+import threading
+
+
+try:
+    import importmagic.index
+    import importmagic.symbols
+    import importmagic.importer
+except ImportError:  # pragma: no cover
+    importmagic = None
+
+
+class ImportMagicError(Exception):
+    """Used to pass defined errors from importmagic to the RPC layer."""
+
+
+class ImportMagic(object):
+
+    def __init__(self):
+        self.is_enabled = bool(importmagic)
+        # fail_message is reported to the user when symbol_index
+        # is (still) None
+        self.fail_message = "symbol index is not yet ready"
+        self.project_root = None
+        self.symbol_index = None
+        self.favorites = set()
+        self._thread = None
+
+    def _build_symbol_index(self, project_root, custom_path, blacklist_re):
+        try:
+            index = importmagic.index.SymbolIndex(blacklist_re=blacklist_re)
+            if os.environ.get('ELPY_TEST'):
+                # test suite support: do not index the whole PYTHONPATH, it
+                # takes much too long
+                index.build_index([])
+            elif custom_path:
+                index.build_index(custom_path)
+            else:
+                index.build_index([project_root] + sys.path)
+        except Exception as e:
+            self.fail_message = "symbol index failed to build: %s" % e
+        else:
+            self.symbol_index = index
+
+    def build_index(self, project_root, custom_path=None, blacklist_re=None):
+        self.project_root = None
+        self._thread = threading.Thread(target=self._build_symbol_index,
+                                        args=(project_root, custom_path,
+                                              blacklist_re))
+        self._thread.setDaemon(True)
+        self._thread.start()
+
+    def get_import_symbols(self, symbol):
+        scores = self.symbol_index.symbol_scores(symbol)
+
+        def sort_key(item):
+            score, mod, var = item
+            if mod in self.favorites:
+                return 2 + score, mod, var
+            return score, mod, var
+
+        scores.sort(key=sort_key, reverse=True)
+        return ["from %s import %s" % (mod, var) if var else "import %s" % mod
+                for (_, mod, var) in scores]
+
+    def add_import(self, source, statement):
+        imports = importmagic.importer.Imports(self.symbol_index, source)
+        if statement.startswith('import '):
+            sepalias = None
+            alias = None
+            if ' as ' in statement:
+                sepalias = statement.find(' as ')
+                alias = statement[sepalias+4:]
+            modname = statement[7:sepalias]
+            imports.add_import(modname, alias)
+            self.favorites.add(modname)
+        else:
+            sep = statement.find(' import ')
+            sepalias = None
+            alias = None
+            if ' as ' in statement:
+                sepalias = statement.find(' as ')
+                alias = statement[sepalias+4:]
+            modname = statement[5:sep]
+            name = statement[sep+8:sepalias]
+            if sep > -1:
+                self.favorites.add(modname)
+                imports.add_import_from(modname, name, alias)
+        start_line, end_line, import_block = imports.get_update()
+        return start_line, end_line, import_block
+
+    def get_unresolved_symbols(self, source):
+        try:
+            scope = importmagic.symbols.Scope.from_source(source)
+        except SyntaxError:
+            raise ImportMagicError('cannot find unresolved names in '
+                                   'incomplete file')
+        unres, unref = scope.find_unresolved_and_unreferenced_symbols()
+        return list(unres)
+
+    def remove_unreferenced_imports(self, source):
+        try:
+            scope = importmagic.symbols.Scope.from_source(source)
+        except SyntaxError:
+            raise ImportMagicError('cannot find unreferenced imports in '
+                                   'incomplete file')
+        unres, unref = scope.find_unresolved_and_unreferenced_symbols()
+        # Note: we do not supply "unres" to the call below, since we do
+        # not want to add imports without querying the user from which
+        # module symbols should be imported.
+        start_line, end_line, import_block = importmagic.importer.get_update(
+            source, self.symbol_index, set(), unref)
+        return start_line, end_line, import_block