Build => get

This commit is contained in:
Luke Van Seters 2021-10-11 15:31:20 -07:00
parent 378f5b248e
commit 9df6dfdf5b

View File

@ -33,15 +33,15 @@ def get_eth_transfers(traces: List[ClassifiedTrace]) -> List[Transfer]:
def get_transfer(trace: ClassifiedTrace) -> Optional[Transfer]:
if trace.value is not None and trace.value > 0:
return _build_eth_transfer(trace)
return _get_eth_transfer(trace)
if isinstance(trace, DecodedCallTrace):
return _build_erc20_transfer(trace)
return _get_erc20_transfer(trace)
return None
def _build_eth_transfer(trace: ClassifiedTrace) -> Transfer:
def _get_eth_transfer(trace: ClassifiedTrace) -> Transfer:
return Transfer(
block_number=trace.block_number,
transaction_hash=trace.transaction_hash,
@ -53,7 +53,7 @@ def _build_eth_transfer(trace: ClassifiedTrace) -> Transfer:
)
def _build_erc20_transfer(trace: DecodedCallTrace) -> Optional[Transfer]:
def _get_erc20_transfer(trace: DecodedCallTrace) -> Optional[Transfer]:
classifier = get_classifier(trace)
if classifier is not None and issubclass(classifier, TransferClassifier):
return classifier.get_transfer(trace)