99
1010
1111class 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