fix: tomllib

maxb2 · May 1, 2023 · 15f53a5 · 15f53a5
1 parent 64868a3
commit 15f53a5
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/typer_config/loaders.py b/typer_config/loaders.py
@@ -6,9 +6,13 @@
 
 from typing import Any, Dict
 
+USING_TOMLLIB = False
+
 try:
     # Only available for python>=3.11
     import tomllib as toml
+
+    USING_TOMLLIB = True
 except ImportError:
     try:
         # Third-party toml parsing library
@@ -95,7 +99,11 @@ def toml_loader(path: str) -> Dict[str, Any]:
         dictionary loaded from file
     """
 
-    with open(path, "r", encoding="utf-8") as _file:
-        conf = toml.load(_file)
+    if USING_TOMLLIB:
+        with open(path, "rb") as _file:
+            conf = toml.load(_file)
+    else:
+        with open(path, "r", encoding="utf-8") as _file:
+            conf = toml.load(_file)
 
     return conf