improv: address Koudai's PR feedback on mypy

Author: heitorlessa

aws_lambda_powertools/utilities/parser/envelopes/base.py

@@ -1,6 +1,6 @@
 import logging
 from abc import ABC, abstractmethod
-from typing import Any, Dict, TypeVar, Union
+from typing import Any, Dict, Optional, TypeVar, Union
 
 from ..types import Model
 
@@ -11,7 +11,7 @@ class BaseEnvelope(ABC):
     """ABC implementation for creating a supported Envelope"""
 
     @staticmethod
-    def _parse(data: Union[Dict[str, Any], str], model: Model) -> Union[Model, None]:
+    def _parse(data: Optional[Union[Dict[str, Any], Any]], model: Model) -> Union[Model, None]:
         """Parses envelope data against model provided
 
         Parameters
@@ -38,7 +38,7 @@ def _parse(data: Union[Dict[str, Any], str], model: Model) -> Union[Model, None]
         return model.parse_obj(data)
 
     @abstractmethod
-    def parse(self, data: Dict[str, Any], model: Model):
+    def parse(self, data: Optional[Union[Dict[str, Any], Any]], model: Model):
         """Implementation to parse data against envelope model, then against the data model
 
         NOTE: Call `_parse` method to fully parse data with model provided.

aws_lambda_powertools/utilities/parser/envelopes/dynamodb.py

@@ -1,7 +1,5 @@
 import logging
-from typing import Any, Dict, List
-
-from typing_extensions import Literal
+from typing import Any, Dict, List, Optional, Union
 
 from ..models import DynamoDBStreamModel
 from ..types import Model
@@ -17,7 +15,7 @@ class DynamoDBStreamEnvelope(BaseEnvelope):
     length of the list is the record's amount in the original event.
     """
 
-    def parse(self, data: Dict[str, Any], model: Model) -> List[Dict[Literal["NewImage", "OldImage"], Model]]:
+    def parse(self, data: Optional[Union[Dict[str, Any], Any]], model: Model) -> List[Dict[str, Optional[Model]]]:
         """Parses DynamoDB Stream records found in either NewImage and OldImage with model provided
 
         Parameters
@@ -30,7 +28,7 @@ def parse(self, data: Dict[str, Any], model: Model) -> List[Dict[Literal["NewIma
         Returns
         -------
         List
-            List of records parsed with model provided
+            List of dictionaries with NewImage and OldImage records parsed with model provided
         """
         logger.debug(f"Parsing incoming data with DynamoDB Stream model {DynamoDBStreamModel}")
         parsed_envelope = DynamoDBStreamModel.parse_obj(data)
@@ -43,5 +41,4 @@ def parse(self, data: Dict[str, Any], model: Model) -> List[Dict[Literal["NewIma
                     "OldImage": self._parse(data=record.dynamodb.OldImage, model=model),
                 }
             )
-        # noinspection PyTypeChecker
         return output

aws_lambda_powertools/utilities/parser/envelopes/event_bridge.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Dict
+from typing import Any, Dict, Optional, Union
 
 from ..models import EventBridgeModel
 from ..types import Model
@@ -11,7 +11,7 @@
 class EventBridgeEnvelope(BaseEnvelope):
     """EventBridge envelope to extract data within detail key"""
 
-    def parse(self, data: Dict[str, Any], model: Model) -> Model:
+    def parse(self, data: Optional[Union[Dict[str, Any], Any]], model: Model) -> Optional[Model]:
         """Parses data found with model provided
 
         Parameters

aws_lambda_powertools/utilities/parser/envelopes/sqs.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Optional, Union
 
 from ..models import SqsModel
 from ..types import Model
@@ -18,7 +18,7 @@ class SqsEnvelope(BaseEnvelope):
     all items in the list will be parsed as str and npt as JSON (and vice versa)
     """
 
-    def parse(self, data: Dict[str, Any], model: Model) -> List[Model]:
+    def parse(self, data: Optional[Union[Dict[str, Any], Any]], model: Model) -> List[Optional[Model]]:
         """Parses records found with model provided
 
         Parameters

aws_lambda_powertools/utilities/parser/parser.py

@@ -12,7 +12,7 @@
 
 @lambda_handler_decorator
 def event_parser(
-    handler: Callable[[Dict, Any], Any],
+    handler: Callable[[Any, LambdaContext], Any],
     event: Dict[str, Any],
     context: LambdaContext,
     model: Model,