Skip to content

Commit c7582f3

Browse files
committed
docs(sqs-base): docstring for base class
1 parent 438ecd5 commit c7582f3

File tree

1 file changed

+13
-4
lines changed
  • aws_lambda_powertools/utilities/batch

1 file changed

+13
-4
lines changed

aws_lambda_powertools/utilities/batch/base.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,10 @@
99

1010

1111
class BaseProcessor(ABC):
12+
"""
13+
Abstract class for batch processors.
14+
"""
15+
1216
@abstractmethod
1317
def _prepare(self):
1418
"""
@@ -24,10 +28,16 @@ def _clean(self):
2428
raise NotImplementedError()
2529

2630
@abstractmethod
27-
def _process_record(self, record):
31+
def _process_record(self, record: Any):
32+
"""
33+
Process record with handler.
34+
"""
2835
raise NotImplementedError()
2936

3037
def process(self) -> List[Tuple]:
38+
"""
39+
Call instance's handler for each record.
40+
"""
3141
return [self._process_record(record) for record in self.records]
3242

3343
def __enter__(self):
@@ -43,7 +53,6 @@ def __call__(self, records: Iterable[Any], handler: Callable):
4353
4454
Parameters
4555
----------
46-
4756
records: Iterable[Any]
4857
Iterable with objects to be processed.
4958
handler: Callable
@@ -59,15 +68,15 @@ class BasePartialProcessor(BaseProcessor):
5968
success_messages: MutableSequence = None
6069
fail_messages: MutableSequence = None
6170

62-
def success_handler(self, record, result):
71+
def success_handler(self, record: Any, result: Any):
6372
"""
6473
Success callback
6574
"""
6675
entry = ("success", result, record)
6776
self.success_messages.append(record)
6877
return entry
6978

70-
def failure_handler(self, record, exception):
79+
def failure_handler(self, record: Any, exception: Exception):
7180
"""
7281
Failure callback
7382
"""

0 commit comments

Comments
 (0)