From f68a06a30d03dbbd7b503f8a3c717ce1f4d8d2d1 Mon Sep 17 00:00:00 2001 From: Aaron Shaw Date: Wed, 16 Oct 2024 01:37:20 +0100 Subject: [PATCH] Update auth.py --- hw_diag/utilities/auth.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/hw_diag/utilities/auth.py b/hw_diag/utilities/auth.py index c7aab99f..11bb10bb 100644 --- a/hw_diag/utilities/auth.py +++ b/hw_diag/utilities/auth.py @@ -1,3 +1,4 @@ +import os import datetime import bcrypt @@ -68,12 +69,19 @@ def write_password(password): def read_password(): + PASSWORD_OVERRIDE = os.getenv('PASSWORD_OVERRIDE', 'false') try: password_row = g.db.query(AuthKeyValue). \ filter(AuthKeyValue.key == 'password_hash'). \ one() except NoResultFound: - default_password = generate_default_password() + if PASSWORD_OVERRIDE != "false": + default_password = PASSWORD_OVERRIDE + logging.info("Using password from override env var!") + else: + default_password = generate_default_password() + logging.info("No password override. Generating default!") + password_row = write_password(default_password) return password_row