Merge pull request #32 from hackebrot/switch-to-tomli

Replace pytoml with tomli and tomli-w 📦

Author: hackebrot

setup.py

@@ -28,7 +28,7 @@ def read(*args: str) -> str:
     include_package_data=True,
     zip_safe=False,
     python_requires=">=3.6",
-    install_requires=["click", "requests", "pytoml", "attrs"],
+    install_requires=["click", "requests", "attrs", "tomli>=1.2.1", "tomli-w>=0.3.0"],
     entry_points={"console_scripts": ["labels = labels.cli:labels"]},
     classifiers=[
         "Development Status :: 3 - Alpha",

src/labels/io.py

@@ -1,7 +1,8 @@
 import logging
 import typing
 
-import pytoml
+import tomli
+import tomli_w
 
 from labels.github import Label
 
@@ -13,16 +14,16 @@ def write_labels(filename: str, labels: typing.List[Label]) -> None:
 
     obj = {label.name: label.params_dict for label in labels}
 
-    with open(filename, "w") as labels_file:
-        pytoml.dump(obj, labels_file)
+    with open(filename, "wb") as labels_file:
+        tomli_w.dump(obj, labels_file)
 
 
 def read_labels(filename: str) -> typing.Dict[str, Label]:
     """Load labels from the given TOML file."""
     logger = logging.getLogger("labels")
     logger.debug(f"Reading labels from {filename}")
 
-    with open(filename, "r") as labels_file:
-        obj = pytoml.load(labels_file)
+    with open(filename, "rb") as labels_file:
+        obj = tomli.load(labels_file)
 
     return {name: Label(**values) for name, values in obj.items()}

tests/test_io.py

@@ -1,6 +1,6 @@
 import typing
 
-import pytoml
+import tomli
 
 from labels.github import Label
 from labels.io import write_labels, read_labels
@@ -14,8 +14,8 @@ def test_write_labels(
     """
     write_labels(labels_file_write, labels)
 
-    with open(labels_file_write, "r") as labels_file:
-        obj = pytoml.load(labels_file)
+    with open(labels_file_write, "rb") as labels_file:
+        obj = tomli.load(labels_file)
 
     assert obj == labels_file_dict