Refactor code to update env_manager import and function name

main
lasseedfast 1 year ago
parent b5b7193a4b
commit dcd3d0a96a
  1. 4
      env_manager/__init__.py
  2. 24
      env_manager/env_manager.py

@ -1,2 +1,2 @@
from .env_manager import get_env from .env_manager import set_env
get_env() set_env()

@ -20,7 +20,7 @@ def get_env():
load_dotenv(".env") load_dotenv(".env")
pwd = os.environ['ARANGO_PWD_ENV_MANAGER'] pwd = os.environ['ARANGO_PWD_ENV_MANAGER']
except (FileNotFoundError, KeyError): except (FileNotFoundError, KeyError):
import getpass from getpass import getpass
import warnings import warnings
warnings.warn("No .env file found or 'ARANGO_PWD_ENV_MANAGER' not set.", UserWarning) 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': ") 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) db = ArangoClient(hosts=host).db(db, username=username, password=pwd)
env_doc = db.collection('enviroment').all().next() env_doc = db.collection('enviroment').all().next()
# Set the environment variables return env_doc
for key, value in env_doc.items():
def set_env():
# Set the environment variables
for key, value in get_env().items():
if isinstance(value, dict): if isinstance(value, dict):
continue continue
os.environ[key] = str(value) os.environ[key] = str(value)
return env_doc 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

Loading…
Cancel
Save