toml support

Author: Richardk2n

pylsp_mypy/plugin.py

@@ -20,6 +20,7 @@
 from pathlib import Path
 from typing import IO, Any, Dict, List, Optional
 
+import toml
 from mypy import api as mypy_api
 from pylsp import hookimpl
 from pylsp.config.config import Config
@@ -324,12 +325,17 @@ def init(workspace: str) -> Dict[str, str]:
     log.info("init workspace = %s", workspace)
 
     configuration = {}
-    path = findConfigFile(workspace, ["pylsp-mypy.cfg", "mypy-ls.cfg", "mypy_ls.cfg"])
+    path = findConfigFile(
+        workspace, ["pylsp-mypy.cfg", "mypy-ls.cfg", "mypy_ls.cfg", "pyproject.toml"]
+    )
     if path:
-        with open(path) as file:
-            configuration = ast.literal_eval(file.read())
+        if "pyproject.toml" in path:
+            configuration = toml.load(path).get("tool").get("pylsp-mypy")
+        else:
+            with open(path) as file:
+                configuration = ast.literal_eval(file.read())
 
-    mypyConfigFile = findConfigFile(workspace, ["mypy.ini", ".mypy.ini"])
+    mypyConfigFile = findConfigFile(workspace, ["mypy.ini", ".mypy.ini", "pyproject.toml"])
     mypyConfigFileMap[workspace] = mypyConfigFile
 
     log.info("mypyConfigFile = %s configuration = %s", mypyConfigFile, configuration)
@@ -368,6 +374,16 @@ def findConfigFile(path: str, names: List[str]) -> Optional[str]:
                             "config file to pylsp-mypy.cfg"
                         )
                     )
+                if file.name == "pyproject.toml":
+                    isPluginConfig = "pylsp-mypy.cfg" in names
+                    configPresent = (
+                        toml.load(file)
+                        .get("tool", {})
+                        .get("pylsp-mypy" if isPluginConfig else "mypy")
+                        is None
+                    )
+                    if not configPresent:
+                        continue
                 return str(file)
 
     return None

pyproject.toml

@@ -16,3 +16,8 @@ exclude = '''
 [tool.isort]
 profile = "black"
 line_length = 100
+
+[tool.pylsp-mypy]
+enabled = true
+live_mode = true
+strict = true

requirements.txt

@@ -1,5 +1,6 @@
 python-lsp-server
 mypy
+toml
 black
 pre-commit
 rstcheck

setup.cfg

@@ -22,6 +22,7 @@ packages = find:
 install_requires =
     python-lsp-server
     mypy
+    toml
 
 
 [options.entry_points]