chore: move generics to types

Author: heitorlessa

aws_lambda_powertools/utilities/parser/parser.py

@@ -1,7 +1,7 @@
 import logging
-from typing import Any, Callable, Dict, Optional, Type, TypeVar, overload
+from typing import Any, Callable, Dict, Optional, Type, overload
 
-from aws_lambda_powertools.utilities.parser.types import Model
+from aws_lambda_powertools.utilities.parser.types import EnvelopeModel, EventParserReturnType, Model
 
 from ...middleware_factory import lambda_handler_decorator
 from ..typing import LambdaContext
@@ -10,8 +10,6 @@
 
 logger = logging.getLogger(__name__)
 
-EventParserReturnType = TypeVar("EventParserReturnType")
-
 
 @lambda_handler_decorator
 def event_parser(
@@ -92,7 +90,7 @@ def parse(event: Dict[str, Any], model: Type[Model]) -> Model:
 
 
 @overload
-def parse(event: Dict[str, Any], model: Type[Model], envelope: Type[Envelope]):
+def parse(event: Dict[str, Any], model: Type[Model], envelope: Type[Envelope]) -> EnvelopeModel:
     ...
 
 

aws_lambda_powertools/utilities/parser/types.py

@@ -12,3 +12,5 @@
     from typing_extensions import Literal  # noqa: F401
 
 Model = TypeVar("Model", bound=BaseModel)
+EnvelopeModel = TypeVar("EnvelopeModel")
+EventParserReturnType = TypeVar("EventParserReturnType")