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

Add Aave transfer classifiers

Browse Source
This commit is contained in:
Gui Heise 2021-10-13 12:57:14 -04:00
parent 7d9d9af120
commit 588b41333d
2 changed files with 27 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mev_inspect/aave_liquidations.py
View File

@ -1,4 +1,4 @@
from typing import List, Tuple from typing import List, Tuple, Optional
from mev_inspect.traces import ( from mev_inspect.traces import (
get_child_traces, get_child_traces,
@ -11,16 +11,9 @@ from mev_inspect.schemas.classified_traces import (
Protocol, Protocol,
) )
<<<<<<< HEAD
<<<<<<< HEAD
=======
from mev_inspect.schemas.transfers import ERC20Transfer, aTokenTransfer, Transfer
>>>>>>> Aave transfers
=======
from mev_inspect.schemas.transfers import ERC20Transfer, Transfer
>>>>>>> Aave transfers
from mev_inspect.schemas.liquidations import Liquidation
from mev_inspect.transfers import get_transfer from mev_inspect.transfers import get_transfer
from mev_inspect.schemas.transfers import Transfer
from mev_inspect.schemas.liquidations import Liquidation
AAVE_CONTRACT_ADDRESSES: List[str] = [ AAVE_CONTRACT_ADDRESSES: List[str] = [
# AAVE Proxy # AAVE Proxy
@ -93,18 +86,18 @@ def _get_payback_token_and_amount(
"""Look for and return liquidator payback from liquidation""" """Look for and return liquidator payback from liquidation"""
child: ClassifiedTrace child: ClassifiedTrace
child_transfer: Transfer child_transfer: Optional[Transfer]
for child in child_traces: for child in child_traces:
if child.classification == Classification.transfer and isinstance( if child.classification == Classification.transfer and isinstance(
child, DecodedCallTrace child, DecodedCallTrace
): ):
child_transfer = ERC20Transfer.from_trace(child) child_transfer = get_transfer(child)
if ( if (
child_transfer.to_address == liquidator child_transfer is not None
and 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

22
mev_inspect/classifiers/specs/aave.py
View File

@ -1,10 +1,28 @@
from mev_inspect.schemas.classified_traces import ( from mev_inspect.schemas.classified_traces import (
Protocol, Protocol,
) )
from mev_inspect.schemas.classifiers import ( from mev_inspect.schemas.classifiers import (
ClassifierSpec, ClassifierSpec,
DecodedCallTrace,
TransferClassifier,
LiquidationClassifier, LiquidationClassifier,
) )
from mev_inspect.schemas.transfers import Transfer
class AaveTransferClassifier(TransferClassifier):
@staticmethod
def get_transfer(trace: DecodedCallTrace) -> Transfer:
return Transfer(
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,
)
AAVE_SPEC = ClassifierSpec( AAVE_SPEC = ClassifierSpec(
@ -19,8 +37,8 @@ ATOKENS_SPEC = ClassifierSpec(
abi_name="aTokens", abi_name="aTokens",
protocol=Protocol.aave, protocol=Protocol.aave,
classifications={ classifications={
"transferOnLiquidation(address,address,uint256)": Classification.transfer, "transferOnLiquidation(address,address,uint256)": AaveTransferClassifier,
"transferFrom(address,address,uint256)": Classification.transfer, "transferFrom(address,address,uint256)": AaveTransferClassifier,
}, },
) )