update batch
This commit is contained in:
@@ -20,8 +20,9 @@ class Status(Enum):
|
||||
|
||||
class Result(NamedTuple):
|
||||
status: Status
|
||||
cause: Any
|
||||
record: Any
|
||||
input_record: Any
|
||||
output: Any | None = None
|
||||
cause: Any | None = None
|
||||
|
||||
|
||||
class BatchProcessor(AbstractContextManager):
|
||||
@@ -112,7 +113,12 @@ class BatchProcessor(AbstractContextManager):
|
||||
result = self.handler(record)
|
||||
|
||||
self.successes.append(record)
|
||||
return Result(Status.SUCCESS, result, record)
|
||||
|
||||
return Result(
|
||||
status=Status.SUCCESS,
|
||||
output=result,
|
||||
input_record=record,
|
||||
)
|
||||
except Exception as exc:
|
||||
exc_str = f'{type(exc).__name__}: {exc}'
|
||||
logger.debug(f'Record processing exception: {exc_str}')
|
||||
@@ -120,4 +126,8 @@ class BatchProcessor(AbstractContextManager):
|
||||
self.exceptions.append(exc)
|
||||
self.failures.append(record)
|
||||
|
||||
return Result(Status.FAIL, exc_str, record)
|
||||
return Result(
|
||||
status=Status.FAIL,
|
||||
input_record=record,
|
||||
cause=exc,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user