Payback function output order

This commit is contained in:
Gui Heise 2021-10-06 13:33:43 -04:00
parent 4c203da24e
commit 4c889f813c

View File

@ -53,9 +53,9 @@ def get_aave_liquidations(
) )
( (
received_amount,
received_token_address, received_token_address,
) = _get_payback_amount_and_token_address(trace, child_traces, liquidator) received_amount,
) = _get_payback_token_and_amount(trace, child_traces, liquidator)
liquidations.append( liquidations.append(
Liquidation( Liquidation(
@ -76,9 +76,9 @@ def get_aave_liquidations(
return liquidations return liquidations
def _get_payback_amount_and_token_address( def _get_payback_token_and_amount(
liquidation: DecodedCallTrace, child_traces: List[ClassifiedTrace], liquidator: str liquidation: DecodedCallTrace, child_traces: List[ClassifiedTrace], liquidator: str
) -> Tuple[int, str]: ) -> Tuple[str, int]:
"""Look for and return liquidator payback from liquidation""" """Look for and return liquidator payback from liquidation"""
for child in child_traces: for child in child_traces:
@ -91,6 +91,6 @@ def _get_payback_amount_and_token_address(
and child_transfer.to_address == liquidator and child_transfer.to_address == liquidator
and child.from_address in AAVE_CONTRACT_ADDRESSES and child.from_address in AAVE_CONTRACT_ADDRESSES
): ):
return child_transfer.amount, child_transfer.token_address return child_transfer.token_address, child_transfer.amount
return 0, liquidation.inputs["_collateral"] return liquidation.inputs["_collateral"], 0