chore: advanced accessing lambda context refactor

Author: rubenfonseca

docs/utilities/batch.md

@@ -273,7 +273,7 @@ Use the context manager to access a list of all returned values from your `recor
 * **When successful**. We will include a tuple with `success`, the result of `record_handler`, and the batch record
 * **When failed**. We will include a tuple with `fail`, exception as a string, and the batch record
 
-```python hl_lines="30-36" title="Accessing processed messages via context manager"
+```python hl_lines="26-34" title="Accessing processed messages via context manager"
 --8<-- "examples/batch_processing/src/context_manager_access.py"
 ```
 
@@ -291,63 +291,14 @@ We can automatically inject the [Lambda context](https://docs.aws.amazon.com/lam
 
 === "As a decorator (legacy)"
 
-    ```python hl_lines="15"
-    from typing import Optional
-
-    from aws_lambda_powertools import Logger, Tracer
-    from aws_lambda_powertools.utilities.batch import (BatchProcessor, EventType,
-                                                       batch_processor)
-    from aws_lambda_powertools.utilities.data_classes.sqs_event import SQSRecord
-    from aws_lambda_powertools.utilities.typing import LambdaContext
-
-    processor = BatchProcessor(event_type=EventType.SQS)
-    tracer = Tracer()
-    logger = Logger()
-
-
-    @tracer.capture_method
-    def record_handler(record: SQSRecord, lambda_context: Optional[LambdaContext] = None):
-        if lambda_context is not None:
-            remaining_time = lambda_context.get_remaining_time_in_millis()
-        ...
-
-
-    @logger.inject_lambda_context
-    @tracer.capture_lambda_handler
-    @batch_processor(record_handler=record_handler, processor=processor)
-    def lambda_handler(event, context: LambdaContext):
-        return processor.response()
+    ```python hl_lines="18"
+    --8<-- "examples/batch_processing/src/advanced_accessing_lambda_context_decorator.py"
     ```
 
 === "As a context manager"
 
-    ```python hl_lines="14 23"
-    from typing import Optional
-
-    from aws_lambda_powertools import Logger, Tracer
-    from aws_lambda_powertools.utilities.batch import BatchProcessor, EventType
-    from aws_lambda_powertools.utilities.data_classes.sqs_event import SQSRecord
-    from aws_lambda_powertools.utilities.typing import LambdaContext
-
-    processor = BatchProcessor(event_type=EventType.SQS)
-    tracer = Tracer()
-    logger = Logger()
-
-
-    @tracer.capture_method
-    def record_handler(record: SQSRecord, lambda_context: Optional[LambdaContext] = None):
-        if lambda_context is not None:
-            remaining_time = lambda_context.get_remaining_time_in_millis()
-        ...
-
-    @logger.inject_lambda_context
-    @tracer.capture_lambda_handler
-    def lambda_handler(event, context: LambdaContext):
-        batch = event["Records"]
-        with processor(records=batch, handler=record_handler, lambda_context=context):
-            result = processor.process()
-
-        return result
+    ```python hl_lines="14 24"
+    --8<-- "examples/batch_processing/src/advanced_accessing_lambda_context_manager.py"
     ```
 
 ### Extending BatchProcessor

examples/batch_processing/src/advanced_accessing_lambda_context.py

@@ -21,7 +21,6 @@ def record_handler(record: SQSRecord, lambda_context: Optional[LambdaContext] =
     if payload:
         item: dict = json.loads(payload)
         logger.info(item)
-    ...
 
 
 @logger.inject_lambda_context

examples/batch_processing/src/advanced_accessing_lambda_context_decorator.py

@@ -0,0 +1,28 @@
+from typing import Optional
+
+from aws_lambda_powertools import Logger, Tracer
+from aws_lambda_powertools.utilities.batch import (
+    BatchProcessor,
+    EventType,
+    batch_processor,
+)
+from aws_lambda_powertools.utilities.data_classes.sqs_event import SQSRecord
+from aws_lambda_powertools.utilities.typing import LambdaContext
+
+processor = BatchProcessor(event_type=EventType.SQS)
+tracer = Tracer()
+logger = Logger()
+
+
+@tracer.capture_method
+def record_handler(record: SQSRecord, lambda_context: Optional[LambdaContext] = None):
+    if lambda_context is not None:
+        remaining_time = lambda_context.get_remaining_time_in_millis()
+        logger.info(remaining_time)
+
+
+@logger.inject_lambda_context
+@tracer.capture_lambda_handler
+@batch_processor(record_handler=record_handler, processor=processor)
+def lambda_handler(event, context: LambdaContext):
+    return processor.response()

examples/batch_processing/src/advanced_accessing_lambda_context_manager.py

@@ -0,0 +1,27 @@
+from typing import Optional
+
+from aws_lambda_powertools import Logger, Tracer
+from aws_lambda_powertools.utilities.batch import BatchProcessor, EventType
+from aws_lambda_powertools.utilities.data_classes.sqs_event import SQSRecord
+from aws_lambda_powertools.utilities.typing import LambdaContext
+
+processor = BatchProcessor(event_type=EventType.SQS)
+tracer = Tracer()
+logger = Logger()
+
+
+@tracer.capture_method
+def record_handler(record: SQSRecord, lambda_context: Optional[LambdaContext] = None):
+    if lambda_context is not None:
+        remaining_time = lambda_context.get_remaining_time_in_millis()
+        logger.info(remaining_time)
+
+
+@logger.inject_lambda_context
+@tracer.capture_lambda_handler
+def lambda_handler(event, context: LambdaContext):
+    batch = event["Records"]
+    with processor(records=batch, handler=record_handler, lambda_context=context):
+        result = processor.process()
+
+    return result