Add interception_response hook

Author: blingblin-g

debug_toolbar/panels/redirects.py

@@ -22,20 +22,8 @@ def _process_response(self, response):
         Common response processing logic.
         """
         if 300 <= response.status_code < 400:
-            redirect_to = response.get("Location")
-            if redirect_to:
-                status_line = f"{response.status_code} {response.reason_phrase}"
-                cookies = response.cookies
-                context = {
-                    "redirect_to": redirect_to,
-                    "status_line": status_line,
-                    "toolbar": self.toolbar,
-                }
-                # Using SimpleTemplateResponse avoids running global context processors.
-                response = SimpleTemplateResponse(
-                    "debug_toolbar/redirect.html", context
-                )
-                response.cookies = cookies
+            if redirect_to := response.get("Location"):
+                response = self.get_interception_response(response, redirect_to)
                 response.render()
         return response
 
@@ -53,3 +41,22 @@ def process_request(self, request):
         if iscoroutine(response):
             return self.aprocess_request(request, response)
         return self._process_response(response)
+
+    def get_interception_response(self, response, redirect_to):
+        """
+        Hook method to allow subclasses to customize the interception response.
+        """
+        status_line = f"{response.status_code} {response.reason_phrase}"
+        cookies = response.cookies
+        original_response = response
+        context = {
+            "redirect_to": redirect_to,
+            "status_line": status_line,
+            "toolbar": self.toolbar,
+            "original_response": original_response,
+        }
+        # Default implementation: return a SimpleTemplateResponse
+        response = SimpleTemplateResponse("debug_toolbar/redirect.html", context)
+        response.cookies = cookies
+        response.original_response = original_response
+        return response

docs/changes.rst

@@ -16,6 +16,7 @@ Pending
   of each target.
 * Avoided reinitializing the staticfiles storage during instrumentation.
 * Fix for exception-unhandled "forked" Promise chain in rebound window.fetch
+* Add hook to RedirectsPanel for subclass customization
 
 5.0.1 (2025-01-13)
 ------------------

tests/panels/test_redirects.py

@@ -85,3 +85,16 @@ async def get_response(request):
         response = await self.panel.process_request(self.request)
         self.assertIsInstance(response, HttpResponse)
         self.assertTrue(response is await_response)
+
+    def test_original_response_preserved(self):
+        redirect = HttpResponse(status=302)
+        redirect["Location"] = "http://somewhere/else/"
+        self._get_response = lambda request: redirect
+        response = self.panel.process_request(self.request)
+        self.assertFalse(response is redirect)
+        self.assertTrue(hasattr(response, "original_response"))
+        self.assertTrue(response.original_response is redirect)
+        self.assertIsNone(response.get("Location"))
+        self.assertEqual(
+            response.original_response.get("Location"), "http://somewhere/else/"
+        )