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 72274fd..1c6f9ae 100644 --- a/mev_inspect/aave_liquidations.py +++ b/mev_inspect/aave_liquidations.py @@ -52,9 +52,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( @@ -75,7 +76,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]: