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

Change classifications => classifiers

Browse Source
This commit is contained in:
Luke Van Seters 2021-10-06 14:53:38 -04:00
parent 8a94eeaf39
commit d2c397f212
10 changed files with 86 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,14 +1,20 @@
from mev_inspect.schemas.classified_traces import (
Classification,
ClassifierSpec,
Protocol,
)
from mev_inspect.schemas.classifiers import (
ClassifierSpec,
Classifier,
)
AAVE_SPEC = ClassifierSpec(
abi_name="AaveLendingPool",
protocol=Protocol.aave,
classifications={
"liquidationCall(address,address,address,uint256,bool)": Classification.liquidate,
classifiers={
"liquidationCall(address,address,address,uint256,bool)": Classifier(
classification=Classification.liquidate,
)
},
)

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

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

5
mev_inspect/classifiers/specs/curve.py
View File

@ -1,8 +1,11 @@
from mev_inspect.schemas.classified_traces import (
ClassifierSpec,
Protocol,
)
from mev_inspect.schemas.classifiers import (
ClassifierSpec,
)
"""
Deployment addresses found here
https://curve.readthedocs.io/ref-addresses.html

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

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

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

@ -1,8 +1,11 @@
from mev_inspect.schemas.classified_traces import (
Classification,
ClassifierSpec,
Protocol,
)
from mev_inspect.schemas.classifiers import (
ClassifierSpec,
Classifier,
)
UNISWAP_V3_CONTRACT_SPECS = [
@ -66,8 +69,10 @@ UNISWAP_V3_CONTRACT_SPECS = [
UNISWAP_V3_GENERAL_SPECS = [
ClassifierSpec(
abi_name="UniswapV3Pool",
classifications={
"swap(address,bool,int256,uint160,bytes)": Classification.swap,
classifiers={
"swap(address,bool,int256,uint160,bytes)": Classifier(
classification=Classification.swap,
)
},
),
ClassifierSpec(
@ -97,8 +102,10 @@ UNISWAPPY_V2_CONTRACT_SPECS = [
UNISWAPPY_V2_PAIR_SPEC = ClassifierSpec(
abi_name="UniswapV2Pair",
classifications={
"swap(uint256,uint256,address,bytes)": Classification.swap,
classifiers={
"swap(uint256,uint256,address,bytes)": Classifier(
classification=Classification.swap,
),
},
)

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

@ -1,16 +1,24 @@
from mev_inspect.schemas.classified_traces import (
Classification,
ClassifierSpec,
Protocol,
)
from mev_inspect.schemas.classifiers import (
ClassifierSpec,
Classifier,
)
WETH_SPEC = ClassifierSpec(
abi_name="WETH9",
protocol=Protocol.weth,
valid_contract_addresses=["0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2"],
classifications={
"transferFrom(address,address,uint256)": Classification.transfer,
"transfer(address,uint256)": Classification.transfer,
classifiers={
"transferFrom(address,address,uint256)": Classifier(
classification=Classification.transfer,
),
"transfer(address,uint256)": Classifier(
classification=Classification.transfer,
),
},
)

4
mev_inspect/classifiers/specs/zero_ex.py
View File

@ -1,7 +1,9 @@
from mev_inspect.schemas.classified_traces import (
ClassifierSpec,
Protocol,
)
from mev_inspect.schemas.classifiers import (
ClassifierSpec,
)
ZEROX_CONTRACT_SPECS = [

7
mev_inspect/classifiers/trace.py
View File

@ -67,8 +67,11 @@ class TraceClassifier:
if call_data is not None:
signature = call_data.function_signature
classification = spec.classifications.get(
signature, Classification.unknown
classifier = spec.classifiers.get(signature)
classification = (
Classification.unknown
if classifier is None
else classifier.classification
)
return DecodedCallTrace(

9
mev_inspect/schemas/classified_traces.py
View File

@ -1,8 +1,6 @@
from enum import Enum
from typing import Any, Dict, List, Optional
from pydantic import BaseModel
from .blocks import Trace
@ -64,10 +62,3 @@ class DecodedCallTrace(CallTrace):
gas_used: Optional[int]
function_name: Optional[str]
function_signature: Optional[str]
class ClassifierSpec(BaseModel):
abi_name: str
protocol: Optional[Protocol] = None
valid_contract_addresses: Optional[List[str]] = None
classifications: Dict[str, Classification] = {}

16
mev_inspect/schemas/classifiers.py Normal file
View File

@ -0,0 +1,16 @@
from typing import Dict, List, Optional
from pydantic import BaseModel
from .classified_traces import Classification, Protocol
class Classifier(BaseModel):
classification: Classification
class ClassifierSpec(BaseModel):
abi_name: str
protocol: Optional[Protocol] = None
valid_contract_addresses: Optional[List[str]] = None
classifiers: Dict[str, Classifier] = {}