zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-24 03:30:50 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update ANDI middleware class

Browse source
This commit is contained in:
Erin Song 2024-08-13 11:40:45 -07:00
parent f1f1ea34ef
commit ebbd329c98
No known key found for this signature in database
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/registrar/registrar_middleware.py
View file

@ -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