refactor(sqs): add module middlewares

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)