def decorator(*args, **kwargs): from celery import current_task tags = {'task_id': current_task.request.id} if 'request_id' in kwargs: tags['request_id'] = kwargs.pop('request_id') with logs.extra_logging(extra=tags): return func(*args, **kwargs)
def test_extra_logging(): with logs.extra_logging({'a': 1}): assert request_context.extra == {'a': 1}
def test_extra_logging(): Context.new() with logs.extra_logging({'a': 1}): assert logs.logging_context.flat == {'a': 1}
def test_extra_logging(): with logs.extra_logging({'a': 1}): assert logs.logging_context.flat == {'a': 1}