From 347f8c9c5c68f9055f0fa0d4fde7db7bad1dabb1 Mon Sep 17 00:00:00 2001 From: Nico Tonnhofer Date: Fri, 23 Feb 2024 10:48:48 +0000 Subject: [PATCH] fix(typing): ensure return type is a str when default_value is set --- .../api_gateway_authorizer_event.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/aws_lambda_powertools/utilities/data_classes/api_gateway_authorizer_event.py b/aws_lambda_powertools/utilities/data_classes/api_gateway_authorizer_event.py index 32d933ecf27..0b1aec43e8a 100644 --- a/aws_lambda_powertools/utilities/data_classes/api_gateway_authorizer_event.py +++ b/aws_lambda_powertools/utilities/data_classes/api_gateway_authorizer_event.py @@ -1,6 +1,6 @@ import enum import re -from typing import Any, Dict, List, Optional +from typing import Any, Dict, List, Optional, overload from aws_lambda_powertools.utilities.data_classes.common import ( BaseRequestContext, @@ -162,6 +162,22 @@ def stage_variables(self) -> Dict[str, str]: def request_context(self) -> BaseRequestContext: return BaseRequestContext(self._data) + @overload + def get_header_value( + self, + name: str, + default_value: str, + case_sensitive: Optional[bool] = False, + ) -> str: ... + + @overload + def get_header_value( + self, + name: str, + default_value: Optional[str] = None, + case_sensitive: Optional[bool] = False, + ) -> Optional[str]: ... + def get_header_value( self, name: str,