Adjust payback function name and migratiop drop to column

This commit is contained in:
Gui Heise 2021-10-06 02:19:13 -04:00
parent d7be215bb9
commit 67c31883c3
2 changed files with 6 additions and 5 deletions

View File

@ -24,4 +24,4 @@ def upgrade():
def downgrade():
op.drop_table("liquidations")
op.drop_column("liquidations", "received_token_address")

View File

@ -51,9 +51,10 @@ def get_aave_liquidations(
trace.transaction_hash, trace.trace_address, traces
)
received_amount, received_token_address = _get_liquidator_payback(
trace, child_traces, liquidator
)
(
received_amount,
received_token_address,
) = _get_payback_amount_and_token_address(trace, child_traces, liquidator)
liquidations.append(
Liquidation(
@ -74,7 +75,7 @@ def get_aave_liquidations(
return liquidations
def _get_liquidator_payback(
def _get_payback_amount_and_token_address(
liquidation: DecodedCallTrace, child_traces: List[ClassifiedTrace], liquidator: str
) -> Tuple[int, str]: