refactor: add load config from

PythiaSocialTech · Oct 10, 2023 · 335b069 · 335b069
1 parent bcff387
commit 335b069
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/sirqle/config.py b/src/sirqle/config.py
@@ -63,3 +63,17 @@ def __init__(
             scheme = str(urlparse(url).scheme)
             if scheme in ["wss", "ws"]:
                 self.client = CLIENT[scheme](url=url)
+        else:
+            url = os.environ.get("SURREAL_URL")
+            namespace = os.environ.get("SURREAL_NS")
+            username = os.environ.get("SURREAL_DB")
+            username = os.environ.get("SURREAL_USER")
+            password = os.environ.get("SURREAL_PASSWORD")
+            scheme = str(urlparse(url).scheme)
+            self.client = CLIENT[scheme](
+                url,
+                namespace=namespace,
+                database=database,
+                username=username,
+                password=password,
+            )