From c823907436ba8b16803ea7433373c259dea04803 Mon Sep 17 00:00:00 2001
From: lasseedfast <lasse.edfast@gmail.com>
Date: Thu, 10 Oct 2024 11:38:54 +0200
Subject: [PATCH] Refactor code to handle missing .env file and set environment
 variable for ArangoDB password

---
 env_manager/env_manager.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/env_manager/env_manager.py b/env_manager/env_manager.py
index 8194207..cb00509 100644
--- a/env_manager/env_manager.py
+++ b/env_manager/env_manager.py
@@ -30,9 +30,11 @@ def get_env():
     try:
         load_dotenv(".env")
         pwd = os.environ['ARANGO_PWD_ENV_MANAGER']
-    except FileNotFoundError:
+    except FileNotFoundError or KeyError:
         print("No .env file found.")
         pwd = input("Enter the ArangoDB password for the user 'env_manager': ")
+        with open(".env", "a+") as f:
+            f.write(f"\nARANGO_PWD_ENV_MANAGER={pwd}")
     host ="https://arango.lasseedfast.se"
     username = "env_manager"
     db='div'