Merge pull request #158 from emcpow2/context-type-hinting

AWS Lambda context type hinting

Author: koxudaxi

mangum/adapter.py

@@ -11,6 +11,8 @@
 from mangum.protocols.http import HTTPCycle
 from mangum.exceptions import ConfigurationError
 
+if typing.TYPE_CHECKING:  # pragma: no cover
+    from awslambdaric.lambda_context import LambdaContext
 
 DEFAULT_TEXT_MIME_TYPES = [
     "application/json",
@@ -79,7 +81,7 @@ def __post_init__(self, text_mime_types: typing.Optional[typing.List[str]]) -> N
             text_mime_types = DEFAULT_TEXT_MIME_TYPES
         self.text_mime_types = text_mime_types
 
-    def __call__(self, event: dict, context: dict) -> dict:
+    def __call__(self, event: dict, context: "LambdaContext") -> dict:
         self.logger.debug("Event received.")
 
         with ExitStack() as stack:

requirements.txt

@@ -10,6 +10,7 @@ moto
 mypy
 brotli
 brotli-asgi
+awslambdaric-stubs
 # Docs
 mkdocs
 mkdocs-material