@@ -384,7 +384,7 @@ def _clean(self):
384
384
raise BatchProcessingError (
385
385
msg = f"All records failed processing. { len (self .exceptions )} individual errors logged"
386
386
f"separately below." ,
387
- child_exceptions = self .exceptions ,
387
+ child_exceptions = tuple ( self .exceptions ) ,
388
388
)
389
389
390
390
messages = self ._get_messages_to_report ()
@@ -411,32 +411,28 @@ def _get_messages_to_report(self) -> Dict[str, str]:
411
411
def _collect_sqs_failures (self ):
412
412
if self .model :
413
413
return {"itemIdentifier" : msg .messageId for msg in self .fail_messages }
414
- else :
415
- return {"itemIdentifier" : msg .message_id for msg in self .fail_messages }
414
+ return {"itemIdentifier" : msg .message_id for msg in self .fail_messages }
416
415
417
416
def _collect_kinesis_failures (self ):
418
417
if self .model :
419
418
# Pydantic model uses int but Lambda poller expects str
420
419
return {"itemIdentifier" : msg .kinesis .sequenceNumber for msg in self .fail_messages }
421
- else :
422
- return {"itemIdentifier" : msg .kinesis .sequence_number for msg in self .fail_messages }
420
+ return {"itemIdentifier" : msg .kinesis .sequence_number for msg in self .fail_messages }
423
421
424
422
def _collect_dynamodb_failures (self ):
425
423
if self .model :
426
424
return {"itemIdentifier" : msg .dynamodb .SequenceNumber for msg in self .fail_messages }
427
- else :
428
- return {"itemIdentifier" : msg .dynamodb .sequence_number for msg in self .fail_messages }
425
+ return {"itemIdentifier" : msg .dynamodb .sequence_number for msg in self .fail_messages }
429
426
430
427
@overload
431
428
def _to_batch_type (self , record : dict , event_type : EventType , model : "BatchTypeModels" ) -> "BatchTypeModels" :
432
- ...
429
+ ... # pragma: no cover
433
430
434
431
@overload
435
432
def _to_batch_type (self , record : dict , event_type : EventType ) -> EventSourceDataClassTypes :
436
- ...
433
+ ... # pragma: no cover
437
434
438
435
def _to_batch_type (self , record : dict , event_type : EventType , model : Optional ["BatchTypeModels" ] = None ):
439
436
if model is not None :
440
437
return model .parse_obj (record )
441
- else :
442
- return self ._DATA_CLASS_MAPPING [event_type ](record )
438
+ return self ._DATA_CLASS_MAPPING [event_type ](record )
0 commit comments