You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
"""Structured JSON logging via structlog."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
import sys
|
|
from contextvars import ContextVar
|
|
from typing import Any
|
|
|
|
import structlog
|
|
|
|
_job_run_id: ContextVar[str] = ContextVar("job_run_id", default="")
|
|
|
|
|
|
def bind_job_run_id(run_id: str) -> None:
|
|
_job_run_id.set(run_id)
|
|
|
|
|
|
def _add_job_run_id(
|
|
logger: Any, method: str, event_dict: dict[str, Any]
|
|
) -> dict[str, Any]:
|
|
run_id = _job_run_id.get()
|
|
if run_id:
|
|
event_dict["job_run_id"] = run_id
|
|
return event_dict
|
|
|
|
|
|
def configure_logging(level: str = "INFO") -> None:
|
|
logging.basicConfig(
|
|
format="%(message)s",
|
|
stream=sys.stdout,
|
|
level=getattr(logging, level.upper(), logging.INFO),
|
|
)
|
|
|
|
structlog.configure(
|
|
processors=[
|
|
structlog.contextvars.merge_contextvars,
|
|
_add_job_run_id,
|
|
structlog.stdlib.add_log_level,
|
|
structlog.processors.TimeStamper(fmt="iso"),
|
|
structlog.processors.StackInfoRenderer(),
|
|
structlog.processors.format_exc_info,
|
|
structlog.processors.JSONRenderer(),
|
|
],
|
|
wrapper_class=structlog.make_filtering_bound_logger(
|
|
getattr(logging, level.upper(), logging.INFO)
|
|
),
|
|
context_class=dict,
|
|
logger_factory=structlog.PrintLoggerFactory(),
|
|
cache_logger_on_first_use=False,
|
|
)
|
|
|
|
|
|
def get_logger(name: str = "") -> structlog.BoundLogger:
|
|
return structlog.get_logger(name)
|