add a check for request attribute in local

namespace-ee · Mar 14, 2024 · 4c80b6f · 4c80b6f
1 parent adb3037
commit 4c80b6f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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