@@ -48,8 +48,8 @@ class EventType(Enum):
48
48
# When using Pydantic Models, it'll accept any
49
49
EventSourceDataClassTypes = Union [SQSRecord , KinesisStreamRecord , DynamoDBRecord ]
50
50
BatchEventTypes = Union [EventSourceDataClassTypes , "BatchTypeModels" ]
51
- SuccessCallback = Tuple [str , Any , BatchEventTypes ]
52
- FailureCallback = Tuple [str , str , BatchEventTypes ]
51
+ SuccessResponse = Tuple [str , Any , BatchEventTypes ]
52
+ FailureResponse = Tuple [str , str , BatchEventTypes ]
53
53
54
54
55
55
class BasePartialProcessor (ABC ):
@@ -111,7 +111,7 @@ def __call__(self, records: List[dict], handler: Callable):
111
111
self .handler = handler
112
112
return self
113
113
114
- def success_handler (self , record , result : Any ) -> SuccessCallback :
114
+ def success_handler (self , record , result : Any ) -> SuccessResponse :
115
115
"""
116
116
Success callback
117
117
@@ -124,7 +124,7 @@ def success_handler(self, record, result: Any) -> SuccessCallback:
124
124
self .success_messages .append (record )
125
125
return entry
126
126
127
- def failure_handler (self , record , exception : _OptExcInfo ) -> FailureCallback :
127
+ def failure_handler (self , record , exception : _OptExcInfo ) -> FailureResponse :
128
128
"""
129
129
Failure callback
130
130
@@ -228,7 +228,7 @@ def _prepare(self):
228
228
self .fail_messages .clear ()
229
229
self .batch_response = self .DEFAULT_RESPONSE
230
230
231
- def _process_record (self , record : dict ) -> Union [SuccessCallback , FailureCallback ]:
231
+ def _process_record (self , record : dict ) -> Union [SuccessResponse , FailureResponse ]:
232
232
"""
233
233
Process a record with instance's handler
234
234
0 commit comments