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

AWS Lambda context type hinting
Kludex · Feb 26, 2021 · 70f0750 · 70f0750
2 parents 3c5894f + fe72ca7
commit 70f0750
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/mangum/adapter.py b/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:

diff --git a/requirements.txt b/requirements.txt
@@ -10,6 +10,7 @@ moto
 mypy
 brotli
 brotli-asgi
+awslambdaric-stubs
 # Docs
 mkdocs
 mkdocs-material