refactor(sqs): add module middlewares

Author: gmcrocetti

aws_lambda_powertools/utilities/batch/__init__.py

@@ -3,9 +3,9 @@
 """
 Batch processing utility
 """
-
 from .base import BasePartialProcessor, BaseProcessor
-from .sqs import PartialSQSProcessor, partial_sqs_processor
+from .middlewares import partial_sqs_processor
+from .sqs import PartialSQSProcessor
 
 __all__ = (
     "BaseProcessor",

aws_lambda_powertools/utilities/batch/middlewares.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+
+"""
+Middlewares for batch utilities
+"""
+from typing import Callable, Dict
+
+from aws_lambda_powertools.middleware_factory import lambda_handler_decorator
+
+from .sqs import PartialSQSProcessor
+
+
+@lambda_handler_decorator
+def partial_sqs_processor(
+    handler: Callable, event: Dict, context: Dict, record_handler: Callable, processor: PartialSQSProcessor = None
+):
+    """
+
+    Examples
+    --------
+
+    """
+    records = event["Records"]
+    processor = processor or PartialSQSProcessor()
+
+    with processor(records, record_handler) as ctx:
+        ctx.process()
+
+    return handler(event, context)