From 9e905c83b099f3251005e0787e2c9993b2a677f1 Mon Sep 17 00:00:00 2001 From: lasseedfast Date: Thu, 10 Oct 2024 11:51:17 +0200 Subject: [PATCH] Refactor code to handle missing .env file and set environment variable for ArangoDB password --- env_manager/env_manager.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/env_manager/env_manager.py b/env_manager/env_manager.py index 34c9358..535cb8c 100644 --- a/env_manager/env_manager.py +++ b/env_manager/env_manager.py @@ -1,7 +1,6 @@ import os from arango import ArangoClient from dotenv import load_dotenv -import warnings def get_env(): """ Initializes a connection to an ArangoDB database and retrieves all documents from the 'enviroment' collection. @@ -21,8 +20,10 @@ def get_env(): load_dotenv(".env") pwd = os.environ['ARANGO_PWD_ENV_MANAGER'] except (FileNotFoundError, KeyError): + import getpass + import warnings warnings.warn("No .env file found or 'ARANGO_PWD_ENV_MANAGER' not set.", UserWarning) - pwd = input("Enter the ArangoDB password for the user 'env_manager': ") + pwd = getpass("Enter the ArangoDB password for the user 'env_manager': ") with open(".env", "a+") as f: f.write(f"\nARANGO_PWD_ENV_MANAGER={pwd}") load_dotenv(".env") # Reload the .env file to set the new environment variable