From d37bf8f6e29b18a7f12bdfa81b5b9677a8beb637 Mon Sep 17 00:00:00 2001 From: Luke Van Seters Date: Fri, 17 Sep 2021 09:30:50 -0400 Subject: [PATCH] Add block number to transfers schema --- mev_inspect/schemas/transfers.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mev_inspect/schemas/transfers.py b/mev_inspect/schemas/transfers.py index af14e73..e62d185 100644 --- a/mev_inspect/schemas/transfers.py +++ b/mev_inspect/schemas/transfers.py @@ -6,6 +6,7 @@ from .classified_traces import Classification, ClassifiedTrace, Protocol class Transfer(BaseModel): + block_number: int transaction_hash: str trace_address: List[int] from_address: str @@ -21,6 +22,7 @@ class EthTransfer(Transfer): @classmethod def from_trace(cls, trace: ClassifiedTrace) -> "EthTransfer": return cls( + block_number=trace.block_number, transaction_hash=trace.transaction_hash, trace_address=trace.trace_address, amount=trace.value, @@ -39,6 +41,7 @@ class ERC20Transfer(Transfer): if trace.protocol == Protocol.weth: return cls( + block_number=trace.block_number, transaction_hash=trace.transaction_hash, trace_address=trace.trace_address, amount=trace.inputs["wad"], @@ -48,6 +51,7 @@ class ERC20Transfer(Transfer): ) else: return cls( + block_number=trace.block_number, transaction_hash=trace.transaction_hash, trace_address=trace.trace_address, amount=trace.inputs["amount"],