Global request middleware

src/core/conf/middleware.py

@@ -15,6 +15,7 @@
     "core.middleware.set_user_from_non_django_authentication.JWTAuthMiddleware",
     "core.middleware.set_user_from_non_django_authentication.TokenAuthMiddleware",
     "core.middleware.global_current_user.set_global_user",
+    "core.middleware.global_request.set_global_request",
     "axes.middleware.AxesMiddleware",
 ]
 

src/core/middleware/global_request.py

@@ -0,0 +1,18 @@
+from typing import Callable
+
+from django.http import HttpRequest, HttpResponse
+
+from core.request import set_request, unset_request
+
+
+def set_global_request(get_response: Callable[[HttpRequest], HttpResponse]) -> Callable[[HttpRequest], HttpResponse]:
+    def middleware(request: HttpRequest) -> HttpResponse:
+        set_request(request)  # type: ignore
+
+        response = get_response(request)
+
+        unset_request()
+
+        return response
+
+    return middleware

src/core/request.py

@@ -0,0 +1,25 @@
+from threading import current_thread, local
+
+from rest_framework.request import Request
+
+_thread_locals = local()
+
+
+def get_request() -> Request | None:
+    return getattr(_thread_locals, _thread_key(), None)
+
+
+def set_request(request: Request) -> None:
+    setattr(_thread_locals, _thread_key(), request)
+
+
+def unset_request() -> None:
+    thread_key = _thread_key()
+
+    if hasattr(_thread_locals, thread_key):
+        delattr(_thread_locals, thread_key)
+
+
+def _thread_key() -> str:
+    thread_name = current_thread().name
+    return f"request_{thread_name}"