feat(toml): switch to tomlkit

pyproject.toml

@@ -13,16 +13,15 @@ python = ">=3.8"
 
 PyYAML = { version = ">=6,<7", optional = true }
 
-tomli = { version = ">=2,<3", optional = true }
-tomli-w = { version = ">=1,<2", optional = true }
+tomlkit = { version = ">=0,<1", optional = true }
 
 pydantic = { version = ">=1,<2", optional = true }
 
 attrs = { version = ">=22.2,<23", optional = true }
 
 [tool.poetry.extras]
 yaml = ["PyYAML"]
-toml = ["tomli", "tomli-w"]
+toml = ["tomlkit"]
 pydantic = ["pydantic"]
 attrs = ["attrs"]
 

zenconfig/formats/toml.py

@@ -2,30 +2,24 @@
 from pathlib import Path
 from typing import Any, Dict
 
-import tomli
-import tomli_w
+import tomlkit
 
 from zenconfig.base import BaseConfig, Format
 
 
 @dataclass
 class TOMLFormat(Format):
-    multiline_strings: bool = True
+    sort_keys: bool = False
 
     def load(self, path: Path) -> Dict[str, Any]:
-        return tomli.loads(path.read_text())
+        return tomlkit.loads(path.read_text())
 
     def dump(
         self,
         path: Path,
         config: Dict[str, Any],
     ) -> None:
-        path.write_text(
-            tomli_w.dumps(
-                config,
-                multiline_strings=self.multiline_strings,
-            )
-        )
+        path.write_text(tomlkit.dumps(config, sort_keys=self.sort_keys))
 
 
 BaseConfig.register_format(TOMLFormat(), ".toml")