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

main
lasseedfast 1 year ago
parent 70a6682051
commit 6e9ead6adf
  1. 2
      env_manager/__init__.py
  2. 4
      env_manager/env_manager.py

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

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

Loading…
Cancel
Save