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

env_manager/__init__.py

@@ -1 +1,3 @@
-from .env_manager import get_env
+from .env_manager import get_env
+env = get_env()
+print("Environment variables set from ArangoDB by env_manager.")

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