normalize_asgi_headers -> use Headers

Author: bitterpanda63

aikido_zen/context/asgi/normalize_asgi_headers.py

@@ -1,17 +1,14 @@
 """Mainly exports normalize_asgi_headers"""
 
+from aikido_zen.helpers.headers import Headers
 
-def normalize_asgi_headers(headers):
+
+def normalize_asgi_headers(headers) -> Headers:
     """
     Normalizes headers provided by ASGI :
     Decodes them, uppercase and underscore keys
     """
-    parsed_headers = {}
+    result = Headers()
     for k, v in headers:
-        # Normalizing key : decoding, removing dashes and uppercase
-        key_without_dashes = k.decode("utf-8").replace("-", "_")
-        key_normalized = key_without_dashes.upper()
-        if not key_normalized in parsed_headers:
-            parsed_headers[key_normalized] = list()
-        parsed_headers[key_normalized].append(v.decode("utf-8"))
-    return parsed_headers
+        result.store_header(k.decode("utf-8"), v.decode("utf-8"))
+    return result

aikido_zen/context/wsgi/extract_wsgi_headers.py

@@ -1,4 +1,5 @@
 """Exports function extract_wsgi_headers"""
+
 from aikido_zen.helpers.headers import Headers