From bf62ccfd21444ff6449618a1b6646f51c094ffe0 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 26 Sep 2023 15:35:12 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(auth.py):=20refactor=20get?=
 =?UTF-8?q?=5Fsecret=5Fkey=20method=20for=20improved=20readability=20and?=
 =?UTF-8?q?=20maintainability=20=F0=9F=94=92=20chore(auth.py):=20update=20?=
 =?UTF-8?q?logging=20level=20to=20debug=20for=20CONFIG=5FDIR=20not=20provi?=
 =?UTF-8?q?ded=20case=20in=20get=5Fsecret=5Fkey=20method?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../langflow/services/settings/auth.py        | 39 +++++++++----------
 1 file changed, 18 insertions(+), 21 deletions(-)

diff --git a/src/backend/langflow/services/settings/auth.py b/src/backend/langflow/services/settings/auth.py
index 621d9e973..cff671b84 100644
--- a/src/backend/langflow/services/settings/auth.py
+++ b/src/backend/langflow/services/settings/auth.py
@@ -70,33 +70,30 @@ class AuthSettings(BaseSettings):
 
     @validator("SECRET_KEY", pre=True)
     def get_secret_key(cls, value, values):
-        config_dir: Path = values.get("CONFIG_DIR")
+        config_dir = values.get("CONFIG_DIR")
 
         if not config_dir:
-            logger.warn("No CONFIG_DIR provided; not saving secret key.")
+            logger.debug("No CONFIG_DIR provided, not saving secret key")
             return value or secrets.token_urlsafe(32)
 
-        secret_key_path = config_dir / "secret_key"
+        secret_key_path = Path(config_dir) / "secret_key"
 
-        try:
-            if value:
-                logger.info("Secret key provided.")
-                write_secret_to_file(secret_key_path, value)
-            else:
-                logger.info("No secret key provided, generating a random one.")
-                if secret_key_path.exists():
-                    value = read_secret_from_file(secret_key_path)
-                    logger.info("Loaded secret key.")
-                    if not value:
-                        value = secrets.token_urlsafe(32)
-                        write_secret_to_file(secret_key_path, value)
-                        logger.info("Saved secret key.")
-                else:
+        if value:
+            logger.debug("Secret key provided")
+            write_secret_to_file(secret_key_path, value)
+        else:
+            logger.debug("No secret key provided, generating a random one")
+
+            if secret_key_path.exists():
+                value = read_secret_from_file(secret_key_path)
+                logger.debug("Loaded secret key")
+                if not value:
                     value = secrets.token_urlsafe(32)
                     write_secret_to_file(secret_key_path, value)
-                    logger.info("Saved secret key.")
-        except IOError:
-            logger.error("Failed to read or write secret key.")
-            raise IOError("Failed to read or write secret key.")
+                    logger.debug("Saved secret key")
+            else:
+                value = secrets.token_urlsafe(32)
+                write_secret_to_file(secret_key_path, value)
+                logger.debug("Saved secret key")
 
         return value