From 4c80b6f2d9087531488e766b3fa25874d2388b1e Mon Sep 17 00:00:00 2001 From: ehabel-kady Date: Thu, 14 Mar 2024 13:36:54 +0200 Subject: [PATCH] add a check for `request` attribute in local --- django_datadog_logger/celery.py | 3 ++- django_datadog_logger/middleware/request_id.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) 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