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

Aave transfers

Browse Source
This commit is contained in:
Gui Heise 2021-10-13 00:53:14 -04:00
parent 5e111dd5b2
commit b2de07407c
2 changed files with 7 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
mev_inspect/aave_liquidations.py
View File

@ -11,7 +11,7 @@ from mev_inspect.schemas.classified_traces import (
Protocol, Protocol,
) )
from mev_inspect.schemas.transfers import ERC20Transfer, aTokenTransfer, Transfer from mev_inspect.schemas.transfers import ERC20Transfer, Transfer
from mev_inspect.schemas.liquidations import Liquidation from mev_inspect.schemas.liquidations import Liquidation
AAVE_CONTRACT_ADDRESSES: List[str] = [ AAVE_CONTRACT_ADDRESSES: List[str] = [
@ -91,22 +91,13 @@ def _get_payback_token_and_amount(
if child.classification == Classification.transfer and isinstance( if child.classification == Classification.transfer and isinstance(
child, DecodedCallTrace child, DecodedCallTrace
): ):
if liquidation.inputs["_receiveAToken"]:
child_transfer = aTokenTransfer.from_trace(child) child_transfer = ERC20Transfer.from_trace(child)
if ( if (
child_transfer.to_address == liquidator child_transfer.to_address == liquidator
) and child.from_address in AAVE_CONTRACT_ADDRESSES: and child.from_address in AAVE_CONTRACT_ADDRESSES
return child_transfer.token_address, child_transfer.amount ):
return child_transfer.token_address, child_transfer.amount
else:
child_transfer = ERC20Transfer.from_trace(child)
if (
child_transfer.to_address == liquidator
) and child.from_address in AAVE_CONTRACT_ADDRESSES:
return child_transfer.token_address, child_transfer.amount
return liquidation.inputs["_collateral"], 0 return liquidation.inputs["_collateral"], 0

17
mev_inspect/schemas/transfers.py
View File

@ -5,7 +5,6 @@ from pydantic import BaseModel
from .classified_traces import ( from .classified_traces import (
Classification, Classification,
ClassifiedTrace, ClassifiedTrace,
DecodedCallTrace,
Protocol, Protocol,
) )
@ -36,22 +35,6 @@ class EthTransfer(Transfer):
) )
class aTokenTransfer(Transfer):
token_address: str
@classmethod
def from_trace(cls, trace: DecodedCallTrace) -> "aTokenTransfer":
return cls(
block_number=trace.block_number,
transaction_hash=trace.transaction_hash,
trace_address=trace.trace_address,
amount=trace.inputs["value"],
to_address=trace.inputs["to"],
from_address=trace.inputs["from"],
token_address=trace.to_address,
)
class ERC20Transfer(Transfer): class ERC20Transfer(Transfer):
token_address: str token_address: str