Change log redaction logic.

atlanhq · Dec 31, 2024 · 63639e9 · 63639e9
1 parent 2c4811d
commit 63639e9
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/pyatlan/utils.py b/pyatlan/utils.py
@@ -335,13 +335,16 @@ class AuthorizationFilter(logging.Filter):
     """
 
     def filter(self, record: logging.LogRecord) -> bool:
-        if record.args and hasattr(record.args, "__iter__"):
+        if isinstance(record.args, dict) and "access_token" in record.args:
+            record.args["access_token"] = "***REDACTED***"  # noqa: S105
+        elif record.args and hasattr(record.args, "__iter__"):
             for arg in record.args:
-                if isinstance(arg, dict):
-                    if "headers" in arg and "authorization" in arg["headers"]:
-                        arg["headers"]["authorization"] = "***REDACTED***"
-                    elif "access_token" in arg:
-                        arg["access_token"] = "***REDACTED***"  # noqa: S105
+                if (
+                    isinstance(arg, dict)
+                    and "headers" in arg
+                    and "authorization" in arg["headers"]
+                ):
+                    arg["headers"]["authorization"] = "***REDACTED***"
 
         return True