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

Add specific classifiers for each type

Browse Source
This commit is contained in:
Luke Van Seters 2021-10-06 15:12:44 -04:00
parent 621a2798c8
commit d57a2d021d
6 changed files with 27 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,10 +1,9 @@
from mev_inspect.schemas.classified_traces import ( from mev_inspect.schemas.classified_traces import (
Classification,
Protocol, Protocol,
) )
from mev_inspect.schemas.classifiers import ( from mev_inspect.schemas.classifiers import (
ClassifierSpec, ClassifierSpec,
Classifier, LiquidationClassifier,
) )
@ -12,9 +11,7 @@ AAVE_SPEC = ClassifierSpec(
abi_name="AaveLendingPool", abi_name="AaveLendingPool",
protocol=Protocol.aave, protocol=Protocol.aave,
classifiers={ classifiers={
"liquidationCall(address,address,address,uint256,bool)": Classifier( "liquidationCall(address,address,address,uint256,bool)": LiquidationClassifier(),
classification=Classification.liquidate,
)
}, },
) )

11
mev_inspect/classifiers/specs/balancer.py
View File

@ -1,10 +1,9 @@
from mev_inspect.schemas.classified_traces import ( from mev_inspect.schemas.classified_traces import (
Classification,
Protocol, Protocol,
) )
from mev_inspect.schemas.classifiers import ( from mev_inspect.schemas.classifiers import (
ClassifierSpec, ClassifierSpec,
Classifier, SwapClassifier,
) )
@ -13,12 +12,8 @@ BALANCER_V1_SPECS = [
abi_name="BPool", abi_name="BPool",
protocol=Protocol.balancer_v1, protocol=Protocol.balancer_v1,
classifiers={ classifiers={
"swapExactAmountIn(address,uint256,address,uint256,uint256)": Classifier( "swapExactAmountIn(address,uint256,address,uint256,uint256)": SwapClassifier(),
classification=Classification.swap, "swapExactAmountOut(address,uint256,address,uint256,uint256)": SwapClassifier(),
),
"swapExactAmountOut(address,uint256,address,uint256,uint256)": Classifier(
classification=Classification.swap,
),
}, },
), ),
] ]

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

@ -1,21 +1,14 @@
from mev_inspect.schemas.classified_traces import (
Classification,
)
from mev_inspect.schemas.classifiers import ( from mev_inspect.schemas.classifiers import (
ClassifierSpec, ClassifierSpec,
Classifier, TransferClassifier,
) )
ERC20_SPEC = ClassifierSpec( ERC20_SPEC = ClassifierSpec(
abi_name="ERC20", abi_name="ERC20",
classifiers={ classifiers={
"transferFrom(address,address,uint256)": Classifier( "transferFrom(address,address,uint256)": TransferClassifier(),
classification=Classification.transfer, "transfer(address,uint256)": TransferClassifier(),
),
"transfer(address,uint256)": Classifier(
classification=Classification.transfer,
),
}, },
) )

11
mev_inspect/classifiers/specs/uniswap.py
View File

@ -1,10 +1,9 @@
from mev_inspect.schemas.classified_traces import ( from mev_inspect.schemas.classified_traces import (
Classification,
Protocol, Protocol,
) )
from mev_inspect.schemas.classifiers import ( from mev_inspect.schemas.classifiers import (
ClassifierSpec, ClassifierSpec,
Classifier, SwapClassifier,
) )
@ -70,9 +69,7 @@ UNISWAP_V3_GENERAL_SPECS = [
ClassifierSpec( ClassifierSpec(
abi_name="UniswapV3Pool", abi_name="UniswapV3Pool",
classifiers={ classifiers={
"swap(address,bool,int256,uint160,bytes)": Classifier( "swap(address,bool,int256,uint160,bytes)": SwapClassifier(),
classification=Classification.swap,
)
}, },
), ),
ClassifierSpec( ClassifierSpec(
@ -103,9 +100,7 @@ UNISWAPPY_V2_CONTRACT_SPECS = [
UNISWAPPY_V2_PAIR_SPEC = ClassifierSpec( UNISWAPPY_V2_PAIR_SPEC = ClassifierSpec(
abi_name="UniswapV2Pair", abi_name="UniswapV2Pair",
classifiers={ classifiers={
"swap(uint256,uint256,address,bytes)": Classifier( "swap(uint256,uint256,address,bytes)": SwapClassifier(),
classification=Classification.swap,
),
}, },
) )

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

@ -1,10 +1,9 @@
from mev_inspect.schemas.classified_traces import ( from mev_inspect.schemas.classified_traces import (
Classification,
Protocol, Protocol,
) )
from mev_inspect.schemas.classifiers import ( from mev_inspect.schemas.classifiers import (
ClassifierSpec, ClassifierSpec,
Classifier, TransferClassifier,
) )
@ -13,12 +12,8 @@ WETH_SPEC = ClassifierSpec(
protocol=Protocol.weth, protocol=Protocol.weth,
valid_contract_addresses=["0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2"], valid_contract_addresses=["0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2"],
classifiers={ classifiers={
"transferFrom(address,address,uint256)": Classifier( "transferFrom(address,address,uint256)": TransferClassifier(),
classification=Classification.transfer, "transfer(address,uint256)": TransferClassifier(),
),
"transfer(address,uint256)": Classifier(
classification=Classification.transfer,
),
}, },
) )

13
mev_inspect/schemas/classifiers.py
View File

@ -1,4 +1,5 @@
from typing import Dict, List, Optional from typing import Dict, List, Optional
from typing_extensions import Literal
from pydantic import BaseModel from pydantic import BaseModel
@ -9,6 +10,18 @@ class Classifier(BaseModel):
classification: Classification classification: Classification
class TransferClassifier(Classifier):
classification: Literal[Classification.transfer] = Classification.transfer
class SwapClassifier(Classifier):
classification: Literal[Classification.swap] = Classification.swap
class LiquidationClassifier(Classifier):
classification: Literal[Classification.liquidate] = Classification.liquidate
class ClassifierSpec(BaseModel): class ClassifierSpec(BaseModel):
abi_name: str abi_name: str
protocol: Optional[Protocol] = None protocol: Optional[Protocol] = None