From 67c31883c3fd4a73c5c0e4255b3f1e45b37221a4 Mon Sep 17 00:00:00 2001 From: Gui Heise Date: Wed, 6 Oct 2021 02:19:13 -0400 Subject: [PATCH] Adjust payback function name and migratiop drop to column --- ...b3_add_received_collateral_address_to_liquidations.py | 2 +- mev_inspect/aave_liquidations.py | 9 +++++---- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/alembic/versions/205ce02374b3_add_received_collateral_address_to_liquidations.py b/alembic/versions/205ce02374b3_add_received_collateral_address_to_liquidations.py index b099292..1a82a2c 100644 --- a/alembic/versions/205ce02374b3_add_received_collateral_address_to_liquidations.py +++ b/alembic/versions/205ce02374b3_add_received_collateral_address_to_liquidations.py @@ -24,4 +24,4 @@ def upgrade(): def downgrade(): - op.drop_table("liquidations") + op.drop_column("liquidations", "received_token_address") diff --git a/mev_inspect/aave_liquidations.py b/mev_inspect/aave_liquidations.py index 29116d6..010807c 100644 --- a/mev_inspect/aave_liquidations.py +++ b/mev_inspect/aave_liquidations.py @@ -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]: