mirror of
https://github.com/apidoorman/doorman.git
synced 2026-05-02 06:09:40 -05:00
seniorswe
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import logging
|
|
import re
|
|
import sys
|
|
|
|
|
|
class _RedactFilter(logging.Filter):
|
|
PATTERNS = [
|
|
re.compile(r'(?i)(authorization\s*[:=]\s*)([^;\r\n]+)'),
|
|
re.compile(r'(?i)(access[_-]?token\s*[\"\']?\s*[:=]\s*[\"\'])([^\"\']+)([\"\'])'),
|
|
re.compile(r'(?i)(refresh[_-]?token\s*[\"\']?\s*[:=]\s*[\"\'])([^\"\']+)([\"\'])'),
|
|
re.compile(r'(?i)(password\s*[\"\']?\s*[:=]\s*[\"\'])([^\"\']+)([\"\'])'),
|
|
re.compile(r'(?i)(cookie\s*[:=]\s*)([^;\r\n]+)'),
|
|
re.compile(r'(?i)(x-csrf-token\s*[:=]\s*)([^\s,;]+)'),
|
|
]
|
|
|
|
def filter(self, record: logging.LogRecord) -> bool:
|
|
try:
|
|
msg = str(record.getMessage())
|
|
red = msg
|
|
for pat in self.PATTERNS:
|
|
red = pat.sub(lambda m: (m.group(1) + '[REDACTED]' + (m.group(3) if m.lastindex and m.lastindex >= 3 else '')), red)
|
|
if red != msg:
|
|
record.msg = red
|
|
except Exception:
|
|
pass
|
|
return True
|
|
|
|
|
|
def _ensure_logger(name: str):
|
|
logger = logging.getLogger(name)
|
|
for h in logger.handlers:
|
|
if h.filters:
|
|
return
|
|
h = logging.StreamHandler(stream=sys.stdout)
|
|
h.setLevel(logging.INFO)
|
|
h.addFilter(_RedactFilter())
|
|
logger.addHandler(h)
|
|
|
|
|
|
try:
|
|
_ensure_logger('doorman.gateway')
|
|
_ensure_logger('doorman.logging')
|
|
except Exception:
|
|
pass
|
|
|