adds tomllib compatibility layer

lazy loads the module
onyx-and-iris · Jul 5, 2023 · ffc5b38 · ffc5b38
1 parent 2ad4aa8
commit ffc5b38
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/slobs_websocket/streamlabs_obs.py b/slobs_websocket/streamlabs_obs.py
@@ -9,7 +9,6 @@
 import time
 from pathlib import Path
 
-import tomli
 import websocket
 
 from . import exceptions
@@ -110,10 +109,15 @@ def __init__(
         self.StreamingService = StreamingService()
 
     def _conn_from_toml(self):
+        try:
+            import tomllib
+        except ModuleNotFoundError:
+            import tomli as tomllib
+
         filepath = Path.cwd() / "config.toml"
         if filepath.is_file():
             with open(filepath, "rb") as f:
-                conn = tomli.load(f)
+                conn = tomllib.load(f)
             return conn["connection"]
 
     def __enter__(self):