get => build

This commit is contained in:
Luke Van Seters 2021-10-12 20:03:08 -04:00
parent 9df6dfdf5b
commit 6de8f494c4

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 _get_eth_transfer(trace)
return build_eth_transfer(trace)
if isinstance(trace, DecodedCallTrace):
return _get_erc20_transfer(trace)
return _build_erc20_transfer(trace)
return None
def _get_eth_transfer(trace: ClassifiedTrace) -> Transfer:
def build_eth_transfer(trace: ClassifiedTrace) -> Transfer:
return Transfer(
block_number=trace.block_number,
transaction_hash=trace.transaction_hash,
@ -53,7 +53,7 @@ def _get_eth_transfer(trace: ClassifiedTrace) -> Transfer:
)
def _get_erc20_transfer(trace: DecodedCallTrace) -> Optional[Transfer]:
def _build_erc20_transfer(trace: DecodedCallTrace) -> Optional[Transfer]:
classifier = get_classifier(trace)
if classifier is not None and issubclass(classifier, TransferClassifier):
return classifier.get_transfer(trace)