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

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

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

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 (
ClassifierSpec,
Classifier,
TransferClassifier,
)
ERC20_SPEC = ClassifierSpec(
abi_name="ERC20",
classifiers={
"transferFrom(address,address,uint256)": Classifier(
classification=Classification.transfer,
),
"transfer(address,uint256)": Classifier(
classification=Classification.transfer,
),
"transferFrom(address,address,uint256)": TransferClassifier(),
"transfer(address,uint256)": TransferClassifier(),
},
)

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

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

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

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

13
mev_inspect/schemas/classifiers.py
View File

@ -1,4 +1,5 @@
from typing import Dict, List, Optional
from typing_extensions import Literal
from pydantic import BaseModel
@ -9,6 +10,18 @@ class Classifier(BaseModel):
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):
abi_name: str
protocol: Optional[Protocol] = None