Skip to content

Commit c2aa872

Browse files
Addressing feedback
1 parent e09fb83 commit c2aa872

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

aws_lambda_powertools/event_handler/api_gateway.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@
5858
VPCLatticeEventV2,
5959
)
6060
from aws_lambda_powertools.utilities.data_classes.common import BaseProxyEvent
61+
from aws_lambda_powertools.utilities.data_classes.shared_functions import get_header_value
6162
from aws_lambda_powertools.utilities.typing import LambdaContext
6263

6364
logger = logging.getLogger(__name__)
@@ -228,12 +229,9 @@ def extract_origin_header(resolver_headers: Dict):
228229
resolver_headers (Dict): A dictionary containing the headers.
229230
230231
Returns:
231-
Union[str, List[str], None]: The value(s) of the 'origin' or 'Origin' header.
232-
If the header is a single header, a string is returned.
233-
If the header is a multi-header, a list of strings is returned.
234-
If the header is not present, None is returned.
232+
Optional[str]: The value(s) of the origin header or None.
235233
"""
236-
resolved_header = resolver_headers.get("origin") or resolver_headers.get("Origin")
234+
resolved_header = get_header_value(resolver_headers, "origin", None, case_sensitive=False)
237235
if isinstance(resolved_header, str):
238236
return resolved_header
239237
if isinstance(resolved_header, list):

0 commit comments

Comments
 (0)