black

python-lsp · Oct 22, 2023 · 6703485 · 6703485
1 parent a442e3b
commit 6703485
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 12 deletions.
diff --git a/pylsp/plugins/rope_autoimport.py b/pylsp/plugins/rope_autoimport.py
@@ -1,6 +1,5 @@
 # Copyright 2022- Python Language Server Contributors.
 
-import code
 import logging
 from typing import Any, Dict, Generator, List, Optional, Set, Union
 
@@ -164,7 +163,7 @@ def _process_statements(
                 "kind": "quickfix",
                 "edit": {"changes": {doc_uri: [edit]}},
                 # data is a supported field for codeAction responses
-                # See https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#textDocument_codeAction
+                # See https://microsoft.github.io/language-server-protocol/specifications/lsp/3.17/specification/#textDocument_codeAction  # pylint: disable=line-too-long
                 "data": {"sortText": _sort_import(score)},
             }
         else:
@@ -202,7 +201,9 @@ def pylsp_completions(
     suggestions = list(autoimport.search_full(word, ignored_names=ignored_names))
     results = list(
         sorted(
-            _process_statements(suggestions, document.uri, word, autoimport, document, "completions"),
+            _process_statements(
+                suggestions, document.uri, word, autoimport, document, "completions"
+            ),
             key=lambda statement: statement["sortText"],
         )
     )
@@ -238,7 +239,7 @@ def pylsp_code_actions(
     config: Config,
     workspace: Workspace,
     document: Document,
-    range: Dict,
+    range: Dict,  # pylint: disable=redefined-builtin
     context: Dict,
 ) -> List[Dict]:
     """
@@ -274,7 +275,14 @@ def pylsp_code_actions(
         log.debug("autoimport: suggestions: %s", suggestions)
         results = list(
             sorted(
-                _process_statements(suggestions, document.uri, word, autoimport, document, "code_actions"),
+                _process_statements(
+                    suggestions,
+                    document.uri,
+                    word,
+                    autoimport,
+                    document,
+                    "code_actions",
+                ),
                 key=lambda statement: statement["data"]["sortText"],
             )
         )

diff --git a/test/plugins/test_autoimport.py b/test/plugins/test_autoimport.py
@@ -38,7 +38,15 @@ def autoimport_workspace(tmp_path_factory) -> Workspace:
         uris.from_fs_path(str(tmp_path_factory.mktemp("pylsp"))), Mock()
     )
     workspace._config = Config(workspace.root_uri, {}, 0, {})
-    workspace._config.update({"rope_autoimport": {"memory": True, "completions": {"enabled": True}, "code_actions": {"enabled": True}}})
+    workspace._config.update(
+        {
+            "rope_autoimport": {
+                "memory": True,
+                "completions": {"enabled": True},
+                "code_actions": {"enabled": True},
+            }
+        }
+    )
     pylsp_initialize(workspace._config, workspace)
     yield workspace
     workspace.close()
@@ -181,9 +189,7 @@ def test_autoimport_code_actions(config, autoimport_workspace):
         config, autoimport_workspace, doc, None, context
     )
     autoimport_workspace.rm_document(DOC_URI)
-    assert any(
-        action.get("title") == "import os" for action in actions
-    )
+    assert any(action.get("title") == "import os" for action in actions)
 
 
 class TestShouldInsert:
@@ -266,9 +272,7 @@ def test_autoimport_completions_for_notebook_document(
                 "plugins": {
                     "rope_autoimport": {
                         "memory": True,
-                        "completions": {
-                            "enabled": True
-                        },
+                        "completions": {"enabled": True},
                     },
                 }
             }