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.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)

View File

@ -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)