WSGI use new headers helper class

Author: bitterpanda63

aikido_zen/context/wsgi/__init__.py

@@ -5,12 +5,13 @@
 from .extract_wsgi_headers import extract_wsgi_headers
 from .build_url_from_wsgi import build_url_from_wsgi
 from ..parse_cookies import parse_cookies
+from ...helpers.headers import Headers
 
 
 @dataclass
 class WSGIContext:
     method: str
-    headers: Dict[str, List[str]]
+    headers: Headers
     cookies: dict
     url: str
     query: dict
@@ -22,7 +23,7 @@ def parse_wsgi_environ(environ) -> WSGIContext:
     This extracts WSGI attributes, described in :
     https://peps.python.org/pep-3333/#environ-variables
     """
-    headers: Dict[str, List[str]] = extract_wsgi_headers(environ)
+    headers: Headers = extract_wsgi_headers(environ)
     # Content type is generally not included as a header, do include this as a header to simplify :
     if "CONTENT_TYPE" in environ:
         headers["CONTENT_TYPE"] = [environ["CONTENT_TYPE"]]

aikido_zen/context/wsgi/extract_wsgi_headers.py

@@ -2,12 +2,15 @@
 
 from typing import Dict, List
 
+from aikido_zen.helpers.headers import Headers
 
-def extract_wsgi_headers(request) -> Dict[str, List[str]]:
+
+def extract_wsgi_headers(request) -> Headers:
     """Extracts WSGI headers which start with HTTP_ from request dict"""
-    headers = {}
+    headers = Headers()
     for key, value in request.items():
         if key.startswith("HTTP_"):
             # Remove the 'HTTP_' prefix and store in the headers dictionary
-            headers[key[5:]] = [value]
+            header_key = key[5:]
+            headers.store_header(header_key, value)
     return headers