a4c93a1df1
improve batch sync add postgres upsert fix sync locking improve listing sync speed up scraper clean up project prepare for github update docker setup
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import logging
|
|
import sys
|
|
from contextvars import ContextVar
|
|
|
|
# Храним trace_id текущего потока/корутины.
|
|
TRACE_ID: ContextVar[str] = ContextVar("trace_id", default="-")
|
|
|
|
|
|
class TraceIdFilter(logging.Filter):
|
|
# Добавляет trace_id в каждую запись лога для сквозной трассировки.
|
|
def filter(self, record: logging.LogRecord) -> bool:
|
|
record.trace_id = TRACE_ID.get()
|
|
return True
|
|
|
|
|
|
def set_trace_id(trace_id: str) -> None:
|
|
TRACE_ID.set(trace_id)
|
|
|
|
|
|
def setup_logging(level: str = "INFO", log_file: str | None = None) -> None:
|
|
handlers: list[logging.Handler] = [logging.StreamHandler(sys.stdout)]
|
|
if log_file:
|
|
handlers.append(logging.FileHandler(log_file, encoding="utf-8"))
|
|
trace_filter = TraceIdFilter()
|
|
for handler in handlers:
|
|
handler.addFilter(trace_filter)
|
|
logging.basicConfig(
|
|
level=getattr(logging, level.upper(), logging.INFO),
|
|
format="%(asctime)s | %(levelname)s | %(name)s | trace=%(trace_id)s | %(message)s",
|
|
handlers=handlers,
|
|
force=True,
|
|
)
|