diff --git a/src/databricks/sql/__init__.py b/src/databricks/sql/__init__.py index c0b6afdc6..9e36582e4 100644 --- a/src/databricks/sql/__init__.py +++ b/src/databricks/sql/__init__.py @@ -30,16 +30,16 @@ def __init__(self): def redact(self, string): return re.sub(self.pattern, self.mask, str(string)) - def filter(self, record): + def filter(self, record: logging.LogRecord): record.msg = self.redact(str(record.msg)) if isinstance(record.args, dict): for k in record.args.keys(): record.args[k] = ( self.redact(record.args[k]) - if isinstance(record.arg[k], str) + if isinstance(record.args[k], str) else record.args[k] ) - else: + elif record.args is not None: record.args = tuple( (self.redact(arg) if isinstance(arg, str) else arg) for arg in record.args