rename get_transfers => get_erc20_transfers

This commit is contained in:
Luke Van Seters 2021-10-08 12:24:43 -04:00
parent afcff7c845
commit b6fc27b3f6
2 changed files with 5 additions and 5 deletions

View File

@ -11,8 +11,8 @@ from mev_inspect.schemas.swaps import Swap
from mev_inspect.schemas.transfers import ERC20Transfer from mev_inspect.schemas.transfers import ERC20Transfer
from mev_inspect.traces import get_traces_by_transaction_hash from mev_inspect.traces import get_traces_by_transaction_hash
from mev_inspect.transfers import ( from mev_inspect.transfers import (
get_transfer,
get_child_transfers, get_child_transfers,
get_erc20_transfer,
filter_transfers, filter_transfers,
remove_child_transfers_of_transfers, remove_child_transfers_of_transfers,
) )
@ -38,7 +38,7 @@ def _get_swaps_for_transaction(traces: List[ClassifiedTrace]) -> List[Swap]:
continue continue
elif trace.classification == Classification.transfer: elif trace.classification == Classification.transfer:
transfer = get_transfer(trace) transfer = get_erc20_transfer(trace)
if transfer is not None: if transfer is not None:
prior_transfers.append(transfer) prior_transfers.append(transfer)

View File

@ -26,14 +26,14 @@ def get_transfers(traces: List[ClassifiedTrace]) -> List[ERC20Transfer]:
for trace in traces: for trace in traces:
if isinstance(trace, DecodedCallTrace): if isinstance(trace, DecodedCallTrace):
transfer = get_transfer(trace) transfer = get_erc20_transfer(trace)
if transfer is not None: if transfer is not None:
transfers.append(transfer) transfers.append(transfer)
return transfers return transfers
def get_transfer(trace: DecodedCallTrace) -> Optional[ERC20Transfer]: def get_erc20_transfer(trace: DecodedCallTrace) -> Optional[ERC20Transfer]:
if not isinstance(trace, DecodedCallTrace): if not isinstance(trace, DecodedCallTrace):
return None return None
@ -55,7 +55,7 @@ def get_child_transfers(
if child_trace.classification == Classification.transfer and isinstance( if child_trace.classification == Classification.transfer and isinstance(
child_trace, DecodedCallTrace child_trace, DecodedCallTrace
): ):
transfer = get_transfer(child_trace) transfer = get_erc20_transfer(child_trace)
if transfer is not None: if transfer is not None:
child_transfers.append(transfer) child_transfers.append(transfer)