diff --git a/django_datadog_logger/celery.py b/django_datadog_logger/celery.py index a022931..fdc8e74 100644 --- a/django_datadog_logger/celery.py +++ b/django_datadog_logger/celery.py @@ -31,7 +31,8 @@ def function_wrapper(*args, **kwargs): local.request = request return func(*args, **kwargs) finally: - del local.request + if hasattr(local, "request"): + del local.request return function_wrapper diff --git a/django_datadog_logger/middleware/request_id.py b/django_datadog_logger/middleware/request_id.py index e1d1da1..9e40ae7 100644 --- a/django_datadog_logger/middleware/request_id.py +++ b/django_datadog_logger/middleware/request_id.py @@ -27,5 +27,6 @@ def __call__(self, request): local.request = request response = self.get_response(request) response["X-Request-ID"] = request.request_id - del local.request + if hasattr(local, "request"): + del local.request return response