44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
import json
|
|
import logging
|
|
from urllib.parse import urlsplit, urlunsplit
|
|
|
|
|
|
SENSITIVE_KEYS = {"url", "location"}
|
|
|
|
|
|
def redact_url(value: str) -> str:
|
|
try:
|
|
parts = urlsplit(value)
|
|
if not parts.scheme or not parts.hostname:
|
|
return "[redacted]"
|
|
host = parts.hostname
|
|
if ":" in host:
|
|
host = f"[{host}]"
|
|
if parts.port:
|
|
host = f"{host}:{parts.port}"
|
|
return urlunsplit((parts.scheme, host, parts.path, "", ""))
|
|
except (TypeError, ValueError):
|
|
return "[redacted]"
|
|
|
|
|
|
class JsonFormatter(logging.Formatter):
|
|
def format(self, record: logging.LogRecord) -> str:
|
|
payload: dict[str, object] = {
|
|
"level": record.levelname,
|
|
"logger": record.name,
|
|
"message": record.getMessage(),
|
|
}
|
|
fields = getattr(record, "fields", {})
|
|
if isinstance(fields, dict):
|
|
for key, value in fields.items():
|
|
payload[key] = redact_url(str(value)) if key in SENSITIVE_KEYS else value
|
|
return json.dumps(payload, default=str, separators=(",", ":"))
|
|
|
|
|
|
def configure_logging(level: str) -> None:
|
|
handler = logging.StreamHandler()
|
|
handler.setFormatter(JsonFormatter())
|
|
root = logging.getLogger()
|
|
root.handlers = [handler]
|
|
root.setLevel(level)
|