Update auth.py

NebraLtd · Oct 16, 2024 · f68a06a · f68a06a
1 parent 01f2208
commit f68a06a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git 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