crowetic/mev-inspect-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add block number to transfers schema

Browse Source
This commit is contained in:
Luke Van Seters 2021-09-17 09:30:50 -04:00
parent f395e9758f
commit d37bf8f6e2
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mev_inspect/schemas/transfers.py
View File

@ -6,6 +6,7 @@ from .classified_traces import Classification, ClassifiedTrace, Protocol
class Transfer(BaseModel): class Transfer(BaseModel):
block_number: int
transaction_hash: str transaction_hash: str
trace_address: List[int] trace_address: List[int]
from_address: str from_address: str
@ -21,6 +22,7 @@ class EthTransfer(Transfer):
@classmethod @classmethod
def from_trace(cls, trace: ClassifiedTrace) -> "EthTransfer": def from_trace(cls, trace: ClassifiedTrace) -> "EthTransfer":
return cls( return cls(
block_number=trace.block_number,
transaction_hash=trace.transaction_hash, transaction_hash=trace.transaction_hash,
trace_address=trace.trace_address, trace_address=trace.trace_address,
amount=trace.value, amount=trace.value,
@ -39,6 +41,7 @@ class ERC20Transfer(Transfer):
if trace.protocol == Protocol.weth: if trace.protocol == Protocol.weth:
return cls( return cls(
block_number=trace.block_number,
transaction_hash=trace.transaction_hash, transaction_hash=trace.transaction_hash,
trace_address=trace.trace_address, trace_address=trace.trace_address,
amount=trace.inputs["wad"], amount=trace.inputs["wad"],
@ -48,6 +51,7 @@ class ERC20Transfer(Transfer):
) )
else: else:
return cls( return cls(
block_number=trace.block_number,
transaction_hash=trace.transaction_hash, transaction_hash=trace.transaction_hash,
trace_address=trace.trace_address, trace_address=trace.trace_address,
amount=trace.inputs["amount"], amount=trace.inputs["amount"],