From 977a72839e2cd8a8c75764ba41be0d7c9dd73271 Mon Sep 17 00:00:00 2001 From: Gui Heise Date: Fri, 7 Jan 2022 11:25:33 -0500 Subject: [PATCH] Remove instance checks --- mev_inspect/aave_liquidations.py | 21 ++++++++------------- 1 file changed, 8 insertions(+), 13 deletions(-) diff --git a/mev_inspect/aave_liquidations.py b/mev_inspect/aave_liquidations.py index e4a595c..05a8de6 100644 --- a/mev_inspect/aave_liquidations.py +++ b/mev_inspect/aave_liquidations.py @@ -2,7 +2,6 @@ from typing import List, Optional, Tuple from mev_inspect.schemas.liquidations import Liquidation from mev_inspect.schemas.traces import ( - CallTrace, Classification, ClassifiedTrace, DecodedCallTrace, @@ -71,15 +70,13 @@ def _get_received_data( for child in child_traces: - if isinstance(child, CallTrace): + child_transfer: Optional[Transfer] = get_transfer(child) - child_transfer: Optional[Transfer] = get_transfer(child) + if child_transfer is not None: - if child_transfer is not None: + if child_transfer.to_address == liquidator: - if child_transfer.to_address == liquidator: - - return child_transfer.token_address, child_transfer.amount + return child_transfer.token_address, child_transfer.amount raise RuntimeError("Transfer from AAVE to liquidator not found!") @@ -91,14 +88,12 @@ def _get_debt_data( for child in child_traces: - if isinstance(child, CallTrace): + child_transfer: Optional[Transfer] = get_transfer(child) - child_transfer: Optional[Transfer] = get_transfer(child) + if child_transfer is not None: - if child_transfer is not None: + if child_transfer.from_address == liquidator: - if child_transfer.from_address == liquidator: - - return child_transfer.token_address, child_transfer.amount + return child_transfer.token_address, child_transfer.amount raise RuntimeError("Transfer from liquidator to AAVE not found!")