1 files changed, 31 insertions, 0 deletions
diff --git a/server/continuedev/core/env.py b/server/continuedev/core/env.py
new file mode 100644
index 00000000..60b86538
--- /dev/null
+++ b/server/continuedev/core/env.py
@@ -0,0 +1,31 @@
+import os
+
+from dotenv import load_dotenv
+
+
+def get_env_var(var_name: str):
+    load_dotenv()
+    return os.getenv(var_name)
+
+
+def make_sure_env_exists():
+    if not os.path.exists(".env"):
+        with open(".env", "w") as f:
+            f.write("")
+
+
+def save_env_var(var_name: str, var_value: str):
+    make_sure_env_exists()
+
+    with open(".env", "r") as f:
+        lines = f.readlines()
+    with open(".env", "w") as f:
+        values = {}
+        for line in lines:
+            key, value = line.split("=")
+            value = value.replace('"', "")
+            values[key] = value
+
+        values[var_name] = var_value
+        for key, value in values.items():
+            f.write(f'{key}="{value}"\n')