Refactor code to handle missing .env file and set environment variable for ArangoDB password

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'