Fix typing in middleware examples

Author: kamilturek

examples/middleware_factory/src/advanced_middleware_tracer_function.py

@@ -14,8 +14,11 @@
 
 
 @lambda_handler_decorator(trace_execution=True)
-def middleware_with_advanced_tracing(handler, event, context) -> Callable:
-
+def middleware_with_advanced_tracing(
+    handler: Callable[[dict, LambdaContext]],
+    event: dict,
+    context: LambdaContext,
+) -> dict:
     tracer.put_metadata(key="resource", value=event.get("resource"))
 
     start_time = time.time()

examples/middleware_factory/src/combining_powertools_utilities_function.py

@@ -26,7 +26,11 @@
 
 
 @lambda_handler_decorator(trace_execution=True)
-def middleware_custom(handler: Callable, event: dict, context: LambdaContext):
+def middleware_custom(
+    handler: Callable[[dict, LambdaContext], dict],
+    event: dict,
+    context: LambdaContext,
+) -> dict:
     # validating the INPUT with the given schema
     # X-Customer-Id header must be informed in all requests
     try:

examples/middleware_factory/src/getting_started_middleware_after_logic_function.py

@@ -12,8 +12,11 @@
 
 
 @lambda_handler_decorator
-def middleware_after(handler, event, context) -> Callable:
-
+def middleware_after(
+    handler: Callable[[dict, LambdaContext], dict],
+    event: dict,
+    context: LambdaContext,
+) -> dict:
     start_time = time.time()
     response = handler(event, context)
     execution_time = time.time() - start_time

examples/middleware_factory/src/getting_started_middleware_before_logic_function.py

@@ -24,7 +24,11 @@ class PaymentError(Exception):
 
 
 @lambda_handler_decorator
-def middleware_before(handler, event, context) -> Callable:
+def middleware_before(
+    handler: Callable[[dict, LambdaContext], dict],
+    event: dict,
+    context: LambdaContext,
+) -> dict:
     # extract payload from a EventBridge event
     detail: dict = extract_data_from_envelope(data=event, envelope=envelopes.EVENTBRIDGE)
 
@@ -38,7 +42,7 @@ def middleware_before(handler, event, context) -> Callable:
 
 
 @middleware_before
-def lambda_handler(event, context: LambdaContext) -> dict:
+def lambda_handler(event: dict, context: LambdaContext) -> dict:
     try:
         payment_payload: dict = extract_data_from_envelope(data=event, envelope=envelopes.EVENTBRIDGE)
         return {

examples/middleware_factory/src/getting_started_middleware_tracer_function.py

@@ -12,8 +12,11 @@
 
 
 @lambda_handler_decorator(trace_execution=True)
-def middleware_with_tracing(handler, event, context) -> Callable:
-
+def middleware_with_tracing(
+    handler: Callable[[dict, LambdaContext], dict],
+    event: dict,
+    context: LambdaContext,
+) -> dict:
     start_time = time.time()
     response = handler(event, context)
     execution_time = time.time() - start_time

examples/middleware_factory/src/getting_started_middleware_with_params_function.py

@@ -28,7 +28,12 @@ class BookingError(Exception):
 
 
 @lambda_handler_decorator
-def obfuscate_sensitive_data(handler, event, context, fields: List) -> Callable:
+def obfuscate_sensitive_data(
+    handler: Callable[[dict, LambdaContext], dict],
+    event: dict,
+    context: LambdaContext,
+    fields: List,
+) -> dict:
     # extracting payload from a EventBridge event
     detail: dict = extract_data_from_envelope(data=event, envelope=envelopes.EVENTBRIDGE)
     guest_data: Any = detail.get("guest")
@@ -49,7 +54,7 @@ def obfuscate_data(value: str) -> bytes:
 
 
 @obfuscate_sensitive_data(fields=["email", "passport", "vat"])
-def lambda_handler(event, context: LambdaContext) -> dict:
+def lambda_handler(event: dict, context: LambdaContext) -> dict:
     try:
         booking_payload: dict = extract_data_from_envelope(data=event, envelope=envelopes.EVENTBRIDGE)
         return {