Merge pull request #45 from flashbots/aave-classifier

Aave classifier
This commit is contained in:
Gui Heise 2021-08-11 10:54:16 -04:00 committed by GitHub
commit d53acded79
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 11 additions and 0 deletions

View File

@ -99,10 +99,19 @@ ERC20_SPEC = ClassifierSpec(
},
)
AAVE_SPEC = ClassifierSpec(
abi_name="AaveLendingPool",
protocol= Protocol.aave,
classifications={
"liquidationCall(address,address,address,uint256,bool)": Classification.liquidate,
},
)
CLASSIFIER_SPECS = [
*UNISWAP_V3_CONTRACT_SPECS,
*UNISWAPPY_V2_CONTRACT_SPECS,
AAVE_SPEC,
ERC20_SPEC,
UNISWAP_V3_POOL_SPEC,
UNISWAPPY_V2_PAIR_SPEC,

View File

@ -11,12 +11,14 @@ class Classification(Enum):
swap = "swap"
burn = "burn"
transfer = "transfer"
liquidate = "liquidate"
class Protocol(Enum):
uniswap_v2 = "uniswap_v2"
uniswap_v3 = "uniswap_v3"
sushiswap = "sushiswap"
aave = "aave"
class ClassifiedTrace(BaseModel):