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)
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,
SwapClassifier,
)
from mev_inspect.classifiers.helpers import get_amount_transferred_to_address
class ZeroExSwapClassifier(SwapClassifier):
@ -40,11 +41,11 @@ class ZeroExSwapClassifier(SwapClassifier):
raise NotImplementedError
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:
if transfer.to_address == taker_address:
token_in_amount = transfer.amount
contract_address = trace.to_address
return Swap(
abi_name=trace.abi_name,