Reverting deletion + adding get_context function

Author: leandrodamascena

aws_lambda_powertools/utilities/data_classes/api_gateway_proxy_event.py

@@ -74,6 +74,16 @@ def route_key(self) -> Optional[str]:
     def authorizer(self) -> APIGatewayEventAuthorizer:
         return APIGatewayEventAuthorizer(self._data["requestContext"]["authorizer"])
 
+    def get_context(self) -> Optional[Dict[str, Any]]:
+        """Retrieve the authorization context details for Lambda.
+
+        Returns:
+        --------
+            Optional[Dict[str, Any]]
+            A dictionary containing Lambda authorization context details, or None if the information is not available.
+        """
+        return self.get("requestContext", {}).get("authorizer", {})
+
 
 class APIGatewayProxyEvent(BaseProxyEvent):
     """AWS Lambda proxy V1
@@ -179,6 +189,21 @@ def jwt_scopes(self) -> Optional[List[str]]:
         jwt = self.get("jwt") or {}  # not available in FunctionURL
         return jwt.get("scopes")
 
+    @property
+    def get_lambda(self) -> Optional[Dict[str, Any]]:
+        """Lambda authorization context details"""
+        return self.get("lambda")
+
+    def get_context(self) -> Optional[Dict[str, Any]]:
+        """Retrieve the authorization context details for Lambda.
+
+        Returns:
+        --------
+            Optional[Dict[str, Any]]
+            A dictionary containing Lambda authorization context details, or None if the information is not available.
+        """
+        return self.get("lambda")
+
     @property
     def iam(self) -> Optional[RequestContextV2AuthorizerIam]:
         """IAM authorization details used for making the request."""

aws_lambda_powertools/utilities/parser/models/apigwv2.py

@@ -1,7 +1,7 @@
 from datetime import datetime
 from typing import Any, Dict, List, Optional, Type, Union
 
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 from pydantic.networks import IPvAnyNetwork
 
 from aws_lambda_powertools.utilities.parser.types import Literal
@@ -31,6 +31,7 @@ class RequestContextV2AuthorizerJwt(BaseModel):
 class RequestContextV2Authorizer(BaseModel):
     jwt: Optional[RequestContextV2AuthorizerJwt] = None
     iam: Optional[RequestContextV2AuthorizerIam] = None
+    lambda_value: Optional[Dict[str, Any]] = Field(None, alias="lambda")
 
 
 class RequestContextV2Http(BaseModel):

tests/events/apiGatewayProxyV2LambdaAuthorizerEvent.json

@@ -32,6 +32,9 @@
                 "scope1",
                 "scope2"
               ]
+            },
+            "lambda": {
+                "tenantId": "123-456-789-012"
             }
           },
         "apiId": "api-id",

tests/unit/data_classes/test_api_gateway_proxy_event.py

@@ -200,6 +200,11 @@ def test_api_gateway_proxy_v2_lambda_authorizer_event():
 
     request_context = parsed_event.request_context
     assert request_context is not None
+
+    lambda_props = request_context.authorizer.get_lambda
+    assert lambda_props is not None
+    assert lambda_props.get("tenantId") == raw_event["requestContext"]["authorizer"]["lambda"]["tenantId"]
+
     jwt_claims = request_context.authorizer.jwt_claim
     assert jwt_claims is not None
     assert jwt_claims.get("claim1") == raw_event["requestContext"]["authorizer"]["jwt"]["claims"]["claim1"]

tests/unit/data_classes/test_lambda_function_url.py

@@ -104,6 +104,7 @@ def test_lambda_function_url_event_iam():
     assert authorizer is not None
     assert authorizer.jwt_claim is None
     assert authorizer.jwt_scopes is None
+    assert authorizer.get_lambda is None
 
     iam = authorizer.iam
     iam_raw = raw_event["requestContext"]["authorizer"]["iam"]

tests/unit/parser/test_apigwv2.py

@@ -77,6 +77,10 @@ def test_api_gateway_proxy_v2_event_lambda_authorizer():
     request_context: RequestContextV2 = parsed_event.requestContext
     assert request_context is not None
 
+    lambda_props: RequestContextV2Authorizer = request_context.authorizer.lambda_value
+    assert lambda_props is not None
+    assert lambda_props["tenantId"] == raw_event["requestContext"]["authorizer"]["lambda"]["tenantId"]
+
     jwt_claims: RequestContextV2Authorizer = request_context.authorizer.jwt.claims
     assert jwt_claims is not None
     assert jwt_claims["claim1"] == raw_event["requestContext"]["authorizer"]["jwt"]["claims"]["claim1"]