Skip to content

Commit 11e385d

Browse files
author
Tom McCarthy
committed
chore: add sqs_batch_processor decorator to simplify interface
1 parent 9c09099 commit 11e385d

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

aws_lambda_powertools/utilities/batch/__init__.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,7 @@
55
"""
66

77
from .base import BasePartialProcessor
8-
from .middlewares import batch_processor
8+
from .middlewares import batch_processor, sqs_batch_processor
99
from .sqs import PartialSQSProcessor
1010

11-
__all__ = (
12-
"BasePartialProcessor",
13-
"PartialSQSProcessor",
14-
"batch_processor",
15-
)
11+
__all__ = ("BasePartialProcessor", "PartialSQSProcessor", "batch_processor", "sqs_batch_processor")

aws_lambda_powertools/utilities/batch/middlewares.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,13 @@
33
"""
44
Middlewares for batch utilities
55
"""
6-
6+
import functools
77
from typing import Callable, Dict
88

99
from aws_lambda_powertools.middleware_factory import lambda_handler_decorator
1010

1111
from .base import BasePartialProcessor
12+
from .sqs import PartialSQSProcessor
1213

1314

1415
@lambda_handler_decorator
@@ -54,3 +55,6 @@ def batch_processor(
5455
processor.process()
5556

5657
return handler(event, context)
58+
59+
60+
sqs_batch_processor = functools.partial(batch_processor, processor=PartialSQSProcessor())

0 commit comments

Comments
 (0)