initial autoimport work

Author: bagel897

pylsp/config/schema.json

@@ -335,6 +335,10 @@
       "type": "boolean",
       "default": true,
       "description": "Enable or disable the plugin."
+    },"pylsp.plugins.rope_autoimport.enabled": {
+      "type": "boolean",
+      "default": true,
+      "description": "Enable or disable the plugin."
     },
     "pylsp.plugins.rope_completion.eager": {
       "type": "boolean",

pylsp/plugins/rope_autoimport.py

@@ -0,0 +1,32 @@
+import logging
+
+from rope.contrib.autoimport import AutoImport
+
+from pylsp import hookimpl, lsp
+from pylsp.config.config import Config
+from pylsp.workspace import Workspace
+
+log = logging.getLogger(__name__)
+
+
+@hookimpl
+def pylsp_settings():
+    # Default rope_completion to disabled
+    return {"plugins": {"rope_autoimport": {"enabled": True}}}
+
+
+@hookimpl
+def pylsp_completions(config: Config, workspace: Workspace, document, position):
+    rope_config = config.settings(document_path=document.path).get("rope", {})
+    rope_project = workspace._rope_project_builder(rope_config)
+    autoimport = AutoImport(rope_project, memory=False)
+    autoimport.close()
+
+
+@hookimpl
+def pylsp_initialize(config: Config, workspace: Workspace):
+    rope_config = config.settings().get("rope", {})
+    rope_project = workspace._rope_project_builder(rope_config)
+    autoimport = AutoImport(rope_project, memory=False)
+    autoimport.generate_modules_cache()
+    autoimport.close()

setup.cfg

@@ -75,6 +75,7 @@ pylsp =
     pylint = pylsp.plugins.pylint_lint
     rope_completion = pylsp.plugins.rope_completion
     rope_rename = pylsp.plugins.rope_rename
+    rope_autoimport = pylsp.plugins.rope_autoimport
     yapf = pylsp.plugins.yapf_format
 
 [pycodestyle]