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

Add transfer classifiers to classifier specs. Implement for erc20 and weth

Browse Source
This commit is contained in:
Luke Van Seters 2021-09-28 13:04:46 -04:00
parent f87016a561
commit 25e1cbcf83
3 changed files with 44 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mev_inspect/classifiers/specs/erc20.py
View File

@ -1,15 +1,34 @@
from typing import Optional
from mev_inspect.schemas.classifier_specs import ClassifierSpec
from mev_inspect.schemas.classified_traces import (
Classification,
DecodedCallTrace,
)
from mev_inspect.schemas.transfers import ERC20Transfer
def classify_transfer(trace: DecodedCallTrace) -> Optional[ERC20Transfer]:
return ERC20Transfer(
block_number=trace.block_number,
transaction_hash=trace.transaction_hash,
trace_address=trace.trace_address,
amount=trace.inputs["amount"],
to_address=trace.inputs["recipient"],
from_address=trace.inputs.get("sender", trace.from_address),
token_address=trace.to_address,
)
ERC20_SPEC = ClassifierSpec(
abi_name="ERC20",
transfer_classifiers={
"transferFrom(address,address,uint256)": classify_transfer,
"transfer(address,uint256)": classify_transfer,
},
classifications={
"transferFrom(address,address,uint256)": Classification.transfer,
"transfer(address,uint256)": Classification.transfer,
"burn(address)": Classification.burn,
},
)

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

@ -1,17 +1,39 @@
from typing import Optional
from mev_inspect.schemas.classifier_specs import ClassifierSpec
from mev_inspect.schemas.classified_traces import (
Classification,
DecodedCallTrace,
Protocol,
)
from mev_inspect.schemas.transfers import ERC20Transfer
def classify_transfer(trace: DecodedCallTrace) -> Optional[ERC20Transfer]:
return ERC20Transfer(
block_number=trace.block_number,
transaction_hash=trace.transaction_hash,
trace_address=trace.trace_address,
amount=trace.inputs["wad"],
to_address=trace.inputs["dst"],
from_address=trace.from_address,
token_address=trace.to_address,
)
WETH_SPEC = ClassifierSpec(
abi_name="WETH9",
protocol=Protocol.weth,
valid_contract_addresses=["0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2"],
transfer_classifiers={
"transferFrom(address,address,uint256)": classify_transfer,
"transfer(address,uint256)": classify_transfer,
},
classifications={
"transferFrom(address,address,uint256)": Classification.transfer,
"transfer(address,uint256)": Classification.transfer,
},
)
WETH_CLASSIFIER_SPECS = [WETH_SPEC]

7
mev_inspect/schemas/classifier_specs.py
View File

@ -3,13 +3,10 @@ from typing import Callable, Dict, List, Optional
from pydantic import BaseModel
from .classified_traces import Classification, DecodedCallTrace, Protocol
from .transfers import Transfer
from .transfers import ERC20Transfer
TransferClassifier = Callable[
[DecodedCallTrace, List[DecodedCallTrace], List[DecodedCallTrace]],
Transfer,
]
TransferClassifier = Callable[[DecodedCallTrace], Optional[ERC20Transfer]]
class ClassifierSpec(BaseModel):