Spaces:
Paused
Paused
File size: 1,111 Bytes
c5fe4a2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import contextvars
import logging
from functools import wraps
from typing import Callable
logging_uuid = contextvars.ContextVar("uuid")
formatter = '%(asctime)s | %(uuid)s [%(pathname)s:%(module)s %(lineno)d] %(levelname)s | %(message)s'
loggingType = logging.CRITICAL | logging.ERROR | logging.WARNING | logging.INFO | logging.DEBUG
def change_logging(level_log: loggingType = logging.INFO) -> None:
old_factory = logging.getLogRecordFactory()
def record_factory(*args, **kwargs):
record = old_factory(*args, **kwargs)
record.uuid = logging_uuid.get("uuid")
if isinstance(record.msg, str):
record.msg = record.msg.replace("\\", "\\\\").replace("\n", "\\n")
return record
logging.setLogRecordFactory(record_factory)
logging.basicConfig(level=level_log, format=formatter, force=True)
def set_uuid_logging(func: Callable) -> Callable:
@wraps(func)
def wrapper(*args, **kwargs):
import uuid
current_uuid = f"{uuid.uuid4()}"
logging_uuid.set(current_uuid)
return func(*args, **kwargs)
return wrapper
|