Merge pull request #4 from aci2n/allow_known_section_config

Allow configuration for known sections

Author: chantera

pylsp_isort/plugin.py

@@ -4,6 +4,7 @@
 from typing import Any, Dict, Generator, Optional, TypedDict, Union
 
 import isort
+from isort.settings import KNOWN_PREFIX
 from pylsp import hookimpl
 from pylsp.config.config import Config
 from pylsp.workspace import Document
@@ -89,7 +90,7 @@ def isort_config(
 
     defined_args = set(getattr(isort.Config, "__dataclass_fields__", {}).keys())
     for key, value in settings.items():
-        if key in defined_args:
+        if key in defined_args or key.startswith(KNOWN_PREFIX):
             config_kwargs[key] = value
         else:
             unsupported_kwargs[key] = value

tests/test_plugin.py

@@ -72,6 +72,20 @@ def test_run_isort(text, settings, expected):
             isort.Config(profile="black"),
             False,
         ),
+        (
+            {
+                "sections": ["FUTURE", "SECTION_A", "SECTION_B"],
+                "known_section_a": ["module_a"],
+                "known_section_b": ["module_b"],
+             },
+            None,
+            isort.Config(
+                sections=["FUTURE", "SECTION_A", "SECTION_B"],
+                known_section_a=["module_a"],
+                known_section_b=["module_b"],
+            ),
+            True,
+        ),
     ],
 )
 def test_isort_config(settings, target_path, expected, check_sources):