Fixed toml parser for style option (#156)

Co-authored-by: Pierre Sassoulas <pierre.sassoulas@gmail.com>
Co-authored-by: Daniël van Noord <13665637+DanielNoord@users.noreply.github.com>

Author: 3 people

pydocstringformatter/_configuration/toml_parsing.py

@@ -45,11 +45,23 @@ def parse_toml_option(
         if value is True:
             return [action.option_strings[0]]
         return []
+
     if isinstance(action, argparse._StoreAction):
         if isinstance(value, int):
             value = str(value)
         return [action.option_strings[0], value]
-    return []  # pragma: no cover
+
+    if isinstance(action, argparse._ExtendAction):  # type: ignore[attr-defined]
+        out_args = []
+        if isinstance(value, list):
+            for item in value:
+                out_args += [action.option_strings[0], item]
+        else:
+            out_args = [action.option_strings[0], value]
+
+        return out_args
+
+    raise NotImplementedError  # pragma: no cover
 
 
 def parse_toml_file(

tests/data/config/valid_toml_numpydoc/pyproject.toml

@@ -0,0 +1,3 @@
+[tool.pydocstringformatter]
+write = false
+style = 'numpydoc'

tests/data/config/valid_toml_numpydoc/test_package/numpydoc_style.py

@@ -0,0 +1,3 @@
+def func():
+    """
+    A docstring"""

tests/data/config/valid_toml_numpydoc_pep257/pyproject.toml

@@ -0,0 +1,3 @@
+[tool.pydocstringformatter]
+write = false
+style = ["numpydoc","pep257"]

tests/data/config/valid_toml_numpydoc_pep257/test_package/numpydoc_style.py

@@ -0,0 +1,3 @@
+def func():
+    """
+    A docstring"""

tests/data/config/valid_toml_pep257/pyproject.toml

@@ -0,0 +1,3 @@
+[tool.pydocstringformatter]
+write = false
+style = 'pep257'

tests/data/config/valid_toml_pep257/test_package/numpydoc_style.py

@@ -0,0 +1,3 @@
+def func():
+    """
+    A docstring"""

tests/test_config.py

@@ -255,3 +255,17 @@ def test_style_invalid_choice(self, capsys: pytest.CaptureFixture[str]) -> None:
         output = capsys.readouterr()
         assert not output.out
         assert "--style: invalid choice: 'invalid'" in output.err
+
+    def test_style_in_toml(self, monkeypatch: pytest.MonkeyPatch) -> None:
+        """Test that the style argument works in the toml file."""
+        monkeypatch.chdir(CONFIG_DATA / "valid_toml_numpydoc")
+        run = _Run(["test_package"])
+        assert ["numpydoc"] == run.config.style
+
+        monkeypatch.chdir(CONFIG_DATA / "valid_toml_pep257")
+        run = _Run(["test_package"])
+        assert ["pep257"] == run.config.style
+
+        monkeypatch.chdir(CONFIG_DATA / "valid_toml_numpydoc_pep257")
+        run = _Run(["test_package"])
+        assert run.config.style == ["numpydoc", "pep257"]