41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
import json
|
|
import logging
|
|
from urllib.parse import urlsplit, urlunsplit
|
|
|
|
_STANDARD = set(logging.makeLogRecord({}).__dict__) | {"message", "asctime"}
|
|
|
|
|
|
def redact_url(value: str) -> str:
|
|
try:
|
|
parsed = urlsplit(value)
|
|
if parsed.scheme not in {"http", "https"} or not parsed.hostname:
|
|
return value
|
|
host = f"[{parsed.hostname}]" if ":" in parsed.hostname else parsed.hostname
|
|
port = f":{parsed.port}" if parsed.port else ""
|
|
query = "<redacted>" if parsed.query else ""
|
|
fragment = "<redacted>" if parsed.fragment else ""
|
|
return urlunsplit((parsed.scheme, host + port, parsed.path, query, fragment))
|
|
except ValueError:
|
|
return "<invalid-url>"
|
|
|
|
|
|
class StructuredJsonFormatter(logging.Formatter):
|
|
def format(self, record: logging.LogRecord) -> str:
|
|
payload: dict[str, object] = {
|
|
"level": record.levelname.lower(),
|
|
"logger": record.name,
|
|
"message": record.getMessage(),
|
|
}
|
|
for key, value in record.__dict__.items():
|
|
if key not in _STANDARD and not key.startswith("_"):
|
|
payload[key] = value
|
|
return json.dumps(payload, default=str, separators=(",", ":"))
|
|
|
|
|
|
def configure_logging(level: str) -> None:
|
|
handler = logging.StreamHandler()
|
|
handler.setFormatter(StructuredJsonFormatter())
|
|
root = logging.getLogger()
|
|
root.handlers[:] = [handler]
|
|
root.setLevel(level.upper())
|