Use spread instead for classified specs list

This commit is contained in:
Luke Van Seters 2021-07-30 14:17:06 -04:00
parent 40529a34bf
commit 127debab18

View File

@ -63,14 +63,12 @@ UNISWAP_V3_CONTRACT_SPECS = [
), ),
] ]
UNISWAP_V3_GENERAL_SPECS = [ UNISWAP_V3_POOL_SPEC = ClassifierSpec(
ClassifierSpec( abi_name="UniswapV3Pool",
abi_name="UniswapV3Pool", classifications={
classifications={ "swap(address,bool,int256,uint160,bytes)": Classification.swap,
"swap(address,bool,int256,uint160,bytes)": Classification.swap, },
}, )
),
]
UNISWAPPY_V2_CONTRACT_SPECS = [ UNISWAPPY_V2_CONTRACT_SPECS = [
ClassifierSpec( ClassifierSpec(
@ -85,14 +83,12 @@ UNISWAPPY_V2_CONTRACT_SPECS = [
), ),
] ]
UNISWAPPY_V2_GENERAL_SPECS = [ UNISWAPPY_V2_PAIR_SPEC = ClassifierSpec(
ClassifierSpec( abi_name="UniswapV2Pair",
abi_name="UniswapV2Pair", classifications={
classifications={ "swap(uint256,uint256,address,bytes)": Classification.swap,
"swap(uint256,uint256,address,bytes)": Classification.swap, },
}, )
),
]
ERC20_SPEC = ClassifierSpec( ERC20_SPEC = ClassifierSpec(
abi_name="ERC20", abi_name="ERC20",
@ -104,10 +100,10 @@ ERC20_SPEC = ClassifierSpec(
) )
CLASSIFIER_SPECS = ( CLASSIFIER_SPECS = [
UNISWAP_V3_CONTRACT_SPECS + *UNISWAP_V3_CONTRACT_SPECS,
UNISWAPPY_V2_CONTRACT_SPECS + *UNISWAPPY_V2_CONTRACT_SPECS,
[ERC20_SPEC] + ERC20_SPEC,
UNISWAP_V3_GENERAL_SPECS + UNISWAP_V3_POOL_SPEC,
UNISWAPPY_V2_GENERAL_SPECS UNISWAPPY_V2_PAIR_SPEC,
) ]