From dcd3d0a96a27feb8b838108ebaf6e1d146bdcf27 Mon Sep 17 00:00:00 2001
From: lasseedfast <lasse.edfast@gmail.com>
Date: Thu, 10 Oct 2024 12:01:45 +0200
Subject: [PATCH] Refactor code to update env_manager import and function name

---
 env_manager/__init__.py    |  4 ++--
 env_manager/env_manager.py | 24 ++++++++++++++++++++----
 2 files changed, 22 insertions(+), 6 deletions(-)

diff --git a/env_manager/__init__.py b/env_manager/__init__.py
index 48c613d..d7a2a75 100644
--- a/env_manager/__init__.py
+++ b/env_manager/__init__.py
@@ -1,2 +1,2 @@
-from .env_manager import get_env
-get_env()
\ No newline at end of file
+from .env_manager import set_env
+set_env()
\ No newline at end of file
diff --git a/env_manager/env_manager.py b/env_manager/env_manager.py
index 535cb8c..6835980 100644
--- a/env_manager/env_manager.py
+++ b/env_manager/env_manager.py
@@ -20,7 +20,7 @@ def get_env():
         load_dotenv(".env")
         pwd = os.environ['ARANGO_PWD_ENV_MANAGER']
     except (FileNotFoundError, KeyError):
-        import getpass
+        from getpass import getpass
         import warnings
         warnings.warn("No .env file found or 'ARANGO_PWD_ENV_MANAGER' not set.", UserWarning)
         pwd = getpass("Enter the ArangoDB password for the user 'env_manager': ")
@@ -37,10 +37,26 @@ def get_env():
     db = ArangoClient(hosts=host).db(db, username=username, password=pwd)
     env_doc = db.collection('enviroment').all().next()
 
-    # Set the environment variables
-    for key, value in env_doc.items():
+    return env_doc
+
+
+def set_env():
+        # Set the environment variables
+    for key, value in get_env().items():
         if isinstance(value, dict):
             continue
         os.environ[key] = str(value)
 
-    return env_doc
\ No newline at end of file
+def print_env():
+        for key, value in get_env().items():
+            if isinstance(value, dict):
+                continue
+            print(f"{key}: {value}")
+
+def print_info():
+    env_info = get_env()['INFO']
+    for i, (k, v) in enumerate(env_info.items()):
+        if i % 2 == 0:
+            print(f"\033[90m{k}: {v}\033[0m")  # Grey color for even indices
+        else:
+            print(f"{k}: {v}")  # Default color for odd indices