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