From 6e9ead6adf087d31f408ae76ea20dce256f3f3df Mon Sep 17 00:00:00 2001 From: lasseedfast Date: Thu, 10 Oct 2024 11:49:37 +0200 Subject: [PATCH] Refactor code to handle missing .env file and set environment variable for ArangoDB password --- env_manager/__init__.py | 4 +++- env_manager/env_manager.py | 4 ++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/env_manager/__init__.py b/env_manager/__init__.py index 9b6805c..91b4fc6 100644 --- a/env_manager/__init__.py +++ b/env_manager/__init__.py @@ -1 +1,3 @@ -from .env_manager import get_env \ No newline at end of file +from .env_manager import get_env +env = get_env() +print("Environment variables set from ArangoDB by env_manager.") \ No newline at end of file diff --git a/env_manager/env_manager.py b/env_manager/env_manager.py index 1201683..34c9358 100644 --- a/env_manager/env_manager.py +++ b/env_manager/env_manager.py @@ -1,7 +1,7 @@ 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,7 +21,7 @@ def get_env(): load_dotenv(".env") pwd = os.environ['ARANGO_PWD_ENV_MANAGER'] except (FileNotFoundError, KeyError): - print("No .env file found or 'ARANGO_PWD_ENV_MANAGER' not set.") + 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': ") with open(".env", "a+") as f: f.write(f"\nARANGO_PWD_ENV_MANAGER={pwd}")