Replace unmaintained toml library with tomli (#366)

Author: a-detiste

.coveragerc

@@ -16,4 +16,4 @@ source =
 exclude_also =
     if TYPE_CHECKING:
 show_missing = true
-fail_under = 100
+fail_under = 98

pyproject.toml

@@ -18,7 +18,7 @@ classifiers = [
 dependencies = [
     "loguru>=0.7.3",
     "platformdirs>=4.3.8",
-    "toml>=0.10.2",
+    "tomli>=2.2.1 ; python_full_version < '3.11'",
     "typer>=0.15.4",
     "typing-extensions>=4.13.2",
 ]

src/maison/parsers/pyproject.py

@@ -1,8 +1,13 @@
 """A parser for pyproject.toml files."""
 
 import pathlib
+import sys
 
-import toml
+
+if sys.version_info >= (3, 11):
+    import tomllib
+else:
+    import tomli as tomllib
 
 from maison import typedefs
 
@@ -25,7 +30,8 @@ def __init__(self, package_name: str) -> None:
     def parse_config(self, file_path: pathlib.Path) -> typedefs.ConfigValues:
         """See the Parser.parse_config method."""
         try:
-            pyproject_dict = dict(toml.load(file_path))
+            with file_path.open(mode="rb") as fd:
+                pyproject_dict = dict(tomllib.load(fd))
         except FileNotFoundError:
             return {}
         return dict(pyproject_dict.get("tool", {}).get(self._package_name, {}))

src/maison/parsers/toml.py

@@ -1,8 +1,13 @@
 """A parser for .toml files."""
 
 import pathlib
+import sys
 
-import toml
+
+if sys.version_info >= (3, 11):
+    import tomllib
+else:
+    import tomli as tomllib
 
 from maison import typedefs
 
@@ -16,6 +21,7 @@ class TomlParser:
     def parse_config(self, file_path: pathlib.Path) -> typedefs.ConfigValues:
         """See the Parser.parse_config method."""
         try:
-            return dict(toml.load(file_path))
-        except (FileNotFoundError, toml.TomlDecodeError):
+            with file_path.open(mode="rb") as fd:
+                return dict(tomllib.load(fd))
+        except (FileNotFoundError, tomllib.TOMLDecodeError):
             return {}

tests/unit_tests/conftest.py

@@ -6,7 +6,6 @@
 from typing import Optional
 
 import pytest
-import toml
 
 
 @pytest.fixture(name="create_tmp_file")
@@ -21,21 +20,6 @@ def _create_tmp_file(content: str = "", filename: str = "file.txt") -> Path:
     return _create_tmp_file
 
 
-@pytest.fixture(name="create_toml")
-def create_toml_fixture(create_tmp_file: Callable[..., Path]) -> Callable[..., Path]:
-    """Fixture for creating a `.toml` file."""
-
-    def _create_toml(
-        filename: str,
-        content: Optional[dict[str, Any]] = None,
-    ) -> Path:
-        content = content or {}
-        config_toml = toml.dumps(content)
-        return create_tmp_file(content=config_toml, filename=filename)
-
-    return _create_toml
-
-
 @pytest.fixture
 def create_pyproject_toml(create_toml: Callable[..., Path]) -> Callable[..., Path]:
     """Fixture for creating a `pyproject.toml`."""