diff --git a/mev_inspect/swaps.py b/mev_inspect/swaps.py index 906759b..ce08770 100644 --- a/mev_inspect/swaps.py +++ b/mev_inspect/swaps.py @@ -11,8 +11,8 @@ from mev_inspect.schemas.swaps import Swap from mev_inspect.schemas.transfers import ERC20Transfer from mev_inspect.traces import get_traces_by_transaction_hash from mev_inspect.transfers import ( - get_transfer, get_child_transfers, + get_erc20_transfer, filter_transfers, remove_child_transfers_of_transfers, ) @@ -38,7 +38,7 @@ def _get_swaps_for_transaction(traces: List[ClassifiedTrace]) -> List[Swap]: continue elif trace.classification == Classification.transfer: - transfer = get_transfer(trace) + transfer = get_erc20_transfer(trace) if transfer is not None: prior_transfers.append(transfer) diff --git a/mev_inspect/transfers.py b/mev_inspect/transfers.py index b1dc11a..aef588f 100644 --- a/mev_inspect/transfers.py +++ b/mev_inspect/transfers.py @@ -26,14 +26,14 @@ def get_transfers(traces: List[ClassifiedTrace]) -> List[ERC20Transfer]: for trace in traces: if isinstance(trace, DecodedCallTrace): - transfer = get_transfer(trace) + transfer = get_erc20_transfer(trace) if transfer is not None: transfers.append(transfer) return transfers -def get_transfer(trace: DecodedCallTrace) -> Optional[ERC20Transfer]: +def get_erc20_transfer(trace: DecodedCallTrace) -> Optional[ERC20Transfer]: if not isinstance(trace, DecodedCallTrace): return None @@ -55,7 +55,7 @@ def get_child_transfers( if child_trace.classification == Classification.transfer and isinstance( child_trace, DecodedCallTrace ): - transfer = get_transfer(child_trace) + transfer = get_erc20_transfer(child_trace) if transfer is not None: child_transfers.append(transfer)