Use document-specific cache proposed in the review

The document identifier is stored in the completion item data.

Author: krassowski

pylsp/hookspecs.py

@@ -30,7 +30,7 @@ def pylsp_completions(config, workspace, document, position):
 
 
 @hookspec(firstresult=True)
-def pylsp_completion_item_resolve(config, workspace, completion_item):
+def pylsp_completion_item_resolve(config, workspace, document, completion_item):
     pass
 
 

pylsp/plugins/jedi_completion.py

@@ -33,22 +33,17 @@
 # Types of parso node for errors
 _ERRORS = ('error_node', )
 
-# most recently retrieved completion items, used for resolution
-_LAST_COMPLETIONS = {}
-
 
 @hookimpl
 def pylsp_completions(config, document, position):
     """Get formatted completions for current code position"""
     # pylint: disable=too-many-locals
-    # pylint: disable=global-statement
-    global _LAST_COMPLETIONS
 
     settings = config.plugin_settings('jedi_completion', document_path=document.path)
     resolve_eagerly = settings.get('eager', False)
     code_position = _utils.position_to_jedi_linecolumn(document, position)
 
-    code_position["fuzzy"] = settings.get("fuzzy", False)
+    code_position['fuzzy'] = settings.get('fuzzy', False)
     completions = document.jedi_script(use_document_path=True).complete(**code_position)
 
     if not completions:
@@ -77,7 +72,13 @@ def pylsp_completions(config, document, position):
                 completion_dict['label'] += ' object'
                 ready_completions.append(completion_dict)
 
-    _LAST_COMPLETIONS = {
+    for completion_dict in ready_completions:
+        completion_dict['data'] = {
+            'doc_uri': document.uri
+        }
+
+    # most recently retrieved completion items, used for resolution
+    document.shared_data['LAST_COMPLETIONS'] = {
         # label is the only required property; here it is assumed to be unique
         completion['label']: (completion, data)
         for completion, data in zip(ready_completions, completions)
@@ -87,9 +88,9 @@ def pylsp_completions(config, document, position):
 
 
 @hookimpl
-def pylsp_completion_item_resolve(completion_item):
+def pylsp_completion_item_resolve(completion_item, document):
     """Resolve formatted completion for given non-resolved completion"""
-    completion, data = _LAST_COMPLETIONS.get(completion_item['label'])
+    completion, data = document.shared_data['LAST_COMPLETIONS'].get(completion_item['label'])
     return _resolve_completion(completion, data)
 
 

pylsp/python_lsp.py

@@ -251,7 +251,8 @@ def completions(self, doc_uri, position):
         }
 
     def completion_item_resolve(self, completion_item):
-        return self._hook('pylsp_completion_item_resolve', completion_item=completion_item)
+        doc_uri = completion_item.get('data', {}).get('doc_uri', None)
+        return self._hook('pylsp_completion_item_resolve', doc_uri, completion_item=completion_item)
 
     def definitions(self, doc_uri, position):
         return flatten(self._hook('pylsp_definitions', doc_uri, position=position))

pylsp/workspace.py

@@ -138,6 +138,7 @@ def __init__(self, uri, workspace, source=None, version=None, local=True, extra_
         self.path = uris.to_fs_path(uri)
         self.dot_path = _utils.path_to_dot_name(self.path)
         self.filename = os.path.basename(self.path)
+        self.shared_data = {}
 
         self._config = workspace._config
         self._workspace = workspace

test/plugins/test_completion.py

@@ -158,7 +158,8 @@ def test_jedi_completion_item_resolve(config, workspace):
     assert 'detail' not in documented_hello_item
 
     resolved_documented_hello = pylsp_jedi_completion_item_resolve(
-        completion_item=documented_hello_item
+        completion_item=documented_hello_item,
+        document=doc
     )
     assert 'Sends a polite greeting' in resolved_documented_hello['documentation']
 
@@ -435,7 +436,7 @@ def test_jedi_completion_environment(workspace):
     completions = pylsp_jedi_completions(doc._config, doc, com_position)
     assert completions[0]['label'] == 'loghub'
 
-    resolved = pylsp_jedi_completion_item_resolve(completions[0])
+    resolved = pylsp_jedi_completion_item_resolve(completions[0], doc)
     assert 'changelog generator' in resolved['documentation'].lower()