diff --git a/env_manager/env_manager.py b/env_manager/env_manager.py index 8194207..cb00509 100644 --- a/env_manager/env_manager.py +++ b/env_manager/env_manager.py @@ -30,9 +30,11 @@ def get_env(): try: load_dotenv(".env") pwd = os.environ['ARANGO_PWD_ENV_MANAGER'] - except FileNotFoundError: + except FileNotFoundError or KeyError: print("No .env file found.") 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}") host ="https://arango.lasseedfast.se" username = "env_manager" db='div'