From ebbd329c986770b498b0a62ed685e39db8351e8e Mon Sep 17 00:00:00 2001
From: Erin Song <121973038+erinysong@users.noreply.github.com>
Date: Tue, 13 Aug 2024 11:40:45 -0700
Subject: [PATCH] Update ANDI middleware class

---
 src/registrar/registrar_middleware.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/registrar/registrar_middleware.py b/src/registrar/registrar_middleware.py
index 2fcaa78b4..fe3c2260d 100644
--- a/src/registrar/registrar_middleware.py
+++ b/src/registrar/registrar_middleware.py
@@ -161,8 +161,15 @@ class CheckPortfolioMiddleware:
 
 
 class ANDIMiddleware(MiddlewareMixin):
-    def process_response(self, request, response):
-        # Check if the response content type is HTML
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request):
+        response = self.get_response(request)
+        return response
+
+    def process_view(self, request, view_func, view_args, view_kwargs):
+        response = self.get_response(request)
         if "text/html" in response.get("Content-Type", ""):
             andi_script = """
             <script src="https://www.ssa.gov/accessibility/andi/andi.js"></script>
@@ -171,4 +178,4 @@ class ANDIMiddleware(MiddlewareMixin):
             content = response.content.decode("utf-8")
             content = content.replace("</body>", f"{andi_script}</body>")
             response.content = content.encode("utf-8")
-        return response
+        return None