fix rope autoimport completion

Author: igmacs

pylsp/plugins/jedi_completion.py

@@ -147,18 +147,27 @@ def pylsp_completion_item_resolve(
     document,
 ):
     """Resolve formatted completion for given non-resolved completion"""
+
+    if (
+        "LAST_JEDI_COMPLETIONS" not in document.shared_data
+        or completion_item["label"] not in document.shared_data["LAST_JEDI_COMPLETIONS"]
+    ):
+        return None
+
     shared_data = document.shared_data["LAST_JEDI_COMPLETIONS"].get(
         completion_item["label"]
     )
 
-    completion_capabilities = config.capabilities.get("textDocument", {}).get(
-        "completion", {}
-    )
-    item_capabilities = completion_capabilities.get("completionItem", {})
-    supported_markup_kinds = item_capabilities.get("documentationFormat", ["markdown"])
-    preferred_markup_kind = _utils.choose_markup_kind(supported_markup_kinds)
-
     if shared_data:
+        completion_capabilities = config.capabilities.get("textDocument", {}).get(
+            "completion", {}
+        )
+        item_capabilities = completion_capabilities.get("completionItem", {})
+        supported_markup_kinds = item_capabilities.get(
+            "documentationFormat", ["markdown"]
+        )
+        preferred_markup_kind = _utils.choose_markup_kind(supported_markup_kinds)
+
         completion, data = shared_data
         return _resolve_completion(
             completion,

pylsp/plugins/rope_autoimport.py

@@ -265,9 +265,31 @@ def pylsp_completions(
     )
     if len(results) > MAX_RESULTS_COMPLETIONS:
         results = results[:MAX_RESULTS_COMPLETIONS]
+
+    # most recently retrieved completion items, used for resolution
+    document.shared_data["LAST_ROPE_AUTOIMPORT_COMPLETIONS"] = {
+        # label is the only required property; here it is assumed to be unique
+        result["label"]: ...
+        for result in results
+    }
+
     return results
 
 
+@hookimpl
+def pylsp_completion_item_resolve(config, completion_item, document):
+    """Resolve formatted completion for given non-resolved completion"""
+
+    if (
+        "LAST_ROPE_AUTOIMPORT_COMPLETIONS" not in document.shared_data
+        or completion_item["label"]
+        not in document.shared_data["LAST_ROPE_AUTOIMPORT_COMPLETIONS"]
+    ):
+        return None
+
+    return completion_item
+
+
 def _document(import_statement: str) -> str:
     return """# Auto-Import\n""" + import_statement
 

pylsp/plugins/rope_completion.py

@@ -92,18 +92,26 @@ def pylsp_completions(config, workspace, document, position):
 @hookimpl
 def pylsp_completion_item_resolve(config, completion_item, document):
     """Resolve formatted completion for given non-resolved completion"""
+
+    if (
+        "LAST_ROPE_COMPLETIONS" not in document.shared_data
+        or completion_item["label"] not in document.shared_data["LAST_ROPE_COMPLETIONS"]
+    ):
+        return None
+
     shared_data = document.shared_data["LAST_ROPE_COMPLETIONS"].get(
         completion_item["label"]
     )
 
-    completion_capabilities = config.capabilities.get("textDocument", {}).get(
-        "completion", {}
-    )
-    item_capabilities = completion_capabilities.get("completionItem", {})
-    supported_markup_kinds = item_capabilities.get("documentationFormat", ["markdown"])
-    preferred_markup_kind = _utils.choose_markup_kind(supported_markup_kinds)
-
     if shared_data:
+        completion_capabilities = config.capabilities.get("textDocument", {}).get(
+            "completion", {}
+        )
+        item_capabilities = completion_capabilities.get("completionItem", {})
+        supported_markup_kinds = item_capabilities.get(
+            "documentationFormat", ["markdown"]
+        )
+        preferred_markup_kind = _utils.choose_markup_kind(supported_markup_kinds)
         completion, data = shared_data
         return _resolve_completion(completion, data, preferred_markup_kind)
     return completion_item