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

Add helper for token_in_amount

Browse Source
This commit is contained in:
Gui Heise 2021-11-22 12:35:23 -05:00
parent f650d3e87f
commit 2f1a9bc751
2 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mev_inspect/classifiers/helpers.py
View File

@ -84,3 +84,10 @@ def _filter_transfers(
filtered_transfers.append(transfer) filtered_transfers.append(transfer)
return filtered_transfers return filtered_transfers
def get_amount_transferred_to_address(address: str, transfers: List[Transfer]) -> int:
for transfer in transfers:
if transfer.to_address == address:
return transfer.amount
return 0

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

@ -9,6 +9,7 @@ from mev_inspect.schemas.classifiers import (
ClassifierSpec, ClassifierSpec,
SwapClassifier, SwapClassifier,
) )
from mev_inspect.classifiers.helpers import get_amount_transferred_to_address
class ZeroExSwapClassifier(SwapClassifier): class ZeroExSwapClassifier(SwapClassifier):
@ -40,11 +41,11 @@ class ZeroExSwapClassifier(SwapClassifier):
raise NotImplementedError raise NotImplementedError
token_out_amount = trace.inputs["takerTokenFillAmount"] token_out_amount = trace.inputs["takerTokenFillAmount"]
contract_address = trace.to_address token_in_amount = get_amount_transferred_to_address(
taker_address, child_transfers
)
for transfer in child_transfers: contract_address = trace.to_address
if transfer.to_address == taker_address:
token_in_amount = transfer.amount
return Swap( return Swap(
abi_name=trace.abi_name, abi_name=trace.abi_name,