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}")