Drop dependency on the toml package

Since we require Python 3.11 or later, we can use the built-in tomllib
module instead.

Signed-off-by: Leandro Lucarella <[email protected]>

Author: llucax

pyproject.toml

@@ -38,7 +38,6 @@ dependencies = [
   "numpy >= 1.24.2, < 2",
   "protobuf >= 4.21.6, < 5",
   "pydantic >= 1.9, < 2",
-  "toml >= 0.10",
   "tqdm >= 4.38.0, < 5",
   "typing_extensions >= 4.4.0, < 5",
   "watchfiles >= 0.15.0",

src/frequenz/sdk/actor/_config_managing.py

@@ -5,9 +5,9 @@
 
 import logging
 import os
+import tomllib
 from typing import Any, Dict, Optional, Set
 
-import toml
 from frequenz.channels import Sender
 from frequenz.channels.util import FileWatcher
 
@@ -58,7 +58,8 @@ def _read_config(self) -> Dict[str, Any]:
             A dictionary containing configuration variables.
         """
         try:
-            return toml.load(self._conf_file)
+            with open(self._conf_file, "rb") as toml_file:
+                return tomllib.load(toml_file)
         except ValueError as err:
             logging.error("Can't read config file, err: %s", err)
             raise

tests/config/test_config.py

@@ -4,12 +4,12 @@
 """Test for Config"""
 import pathlib
 import re
+
+# pylint: disable = no-name-in-module
+import tomllib
 from typing import Any, Dict, List, Optional, Set
 
 import pytest
-
-# pylint: disable = no-name-in-module
-import toml
 from pydantic import BaseModel, StrictBool, StrictFloat, StrictInt
 
 from frequenz.sdk.config import Config
@@ -54,29 +54,32 @@ def config_file(self, tmp_path: pathlib.Path) -> pathlib.Path:
         file_path.write_text(TestConfig.conf_content)
         return file_path
 
-    def test_get(self, config_file: pathlib.Path) -> None:
+    @pytest.fixture()
+    def conf_vars(self, config_file: pathlib.Path) -> dict[str, Any]:
+        """Load the created test config file."""
+        with config_file.open("rb") as file:
+            return tomllib.load(file)
+
+    def test_get(self, conf_vars: dict[str, Any]) -> None:
         """Test get function"""
-        conf_vars = toml.load(config_file)
         config = Config(conf_vars=conf_vars)
 
         assert config.get("logging_lvl") == "DEBUG"
         assert config.get("var1") == "1"
         assert config.get("var2") is None
         assert config.get("var2", default=0) == 0
 
-    def test_getitem(self, config_file: pathlib.Path) -> None:
+    def test_getitem(self, conf_vars: dict[str, Any]) -> None:
         """Test getitem function"""
-        conf_vars = toml.load(config_file)
         config = Config(conf_vars=conf_vars)
 
         assert config["logging_lvl"] == "DEBUG"
         assert config["var1"] == "1"
         with pytest.raises(KeyError, match="Unknown config name var2"):
             assert config["var2"]
 
-    def test_contains(self, config_file: pathlib.Path) -> None:
+    def test_contains(self, conf_vars: dict[str, Any]) -> None:
         """Test contains function"""
-        conf_vars = toml.load(config_file)
         config = Config(conf_vars=conf_vars)
 
         assert "logging_lvl" in config
@@ -111,11 +114,10 @@ def test_contains(self, config_file: pathlib.Path) -> None:
         ],
     )
     def test_get_as_success(
-        self, key: str, expected_type: Any, value: Any, config_file: pathlib.Path
+        self, key: str, expected_type: Any, value: Any, conf_vars: dict[str, Any]
     ) -> None:
         """Test get_as function with proper arguments"""
 
-        conf_vars = toml.load(config_file)
         config = Config(conf_vars=conf_vars)
         result = config.get_as(key, expected_type)
         assert result == value
@@ -132,10 +134,9 @@ def test_get_as_success(
         ],
     )
     def test_get_as_validation_error(
-        self, key: str, expected_type: Any, config_file: pathlib.Path
+        self, key: str, expected_type: Any, conf_vars: dict[str, Any]
     ) -> None:
         """Test get_as function which raise ValidationError"""
-        conf_vars = toml.load(config_file)
         config = Config(conf_vars=conf_vars)
 
         err_msg = (
@@ -158,11 +159,10 @@ def test_get_dict_values_success(
         key_prefix: str,
         expected_values_type: Any,
         value: Any,
-        config_file: pathlib.Path,
+        conf_vars: dict[str, Any],
     ) -> None:
         """Test get_as function with proper arguments"""
 
-        conf_vars = toml.load(config_file)
         config = Config(conf_vars=conf_vars)
         result = config.get_dict(key_prefix, expected_values_type)
         assert result == value
@@ -176,11 +176,10 @@ def test_get_dict_values_success(
         ],
     )
     def test_get_dict_success(
-        self, key_prefix: str, expected_values_type: Any, config_file: pathlib.Path
+        self, key_prefix: str, expected_values_type: Any, conf_vars: dict[str, Any]
     ) -> None:
         """Test get_as function with proper arguments"""
 
-        conf_vars = toml.load(config_file)
         config = Config(conf_vars=conf_vars)
 
         err_msg_re = (