Remove unused logic/ minor fixes

This commit is contained in:
Gui Heise 2021-09-21 08:39:18 -04:00
parent fededa9cad
commit 7f93466b35

View File

@ -5,10 +5,11 @@ from mev_inspect.schemas.classified_traces import (
DecodedCallTrace, DecodedCallTrace,
Classification, Classification,
) )
from mev_inspect.schemas.liquidations import Liquidation from mev_inspect.schemas.liquidations import Liquidation
from mev_inspect.schemas.transfers import ERC20Transfer from mev_inspect.schemas.transfers import ERC20Transfer
contract_addresses = [ LIQUIDATION_CONTRACT_ADDRESSES = [
"0x3dfd23A6c5E8BbcFc9581d2E864a68feb6a076d3", "0x3dfd23A6c5E8BbcFc9581d2E864a68feb6a076d3",
"0x7d2768dE32b0b80b7a3454c06BdAc94A69DDc7A9", "0x7d2768dE32b0b80b7a3454c06BdAc94A69DDc7A9",
"0x398eC7346DcD622eDc5ae82352F02bE94C62d119", "0x398eC7346DcD622eDc5ae82352F02bE94C62d119",
@ -16,7 +17,7 @@ contract_addresses = [
def find_liquidations_from_traces( def find_liquidations_from_traces(
traces: List[ClassifiedTrace], traces: List,
) -> List: ) -> List:
"""Inspect list of classified traces and identify liquidation""" """Inspect list of classified traces and identify liquidation"""
tx = [] tx = []
@ -38,7 +39,7 @@ def find_liquidations_from_traces(
): ):
liquidations.append(trace) liquidations.append(trace)
contract_addresses.append(trace.from_address) LIQUIDATION_CONTRACT_ADDRESSES.append(trace.from_address)
tx.append(trace.transaction_hash) tx.append(trace.transaction_hash)
# Found liquidation, now parse inputs for data # Found liquidation, now parse inputs for data
@ -70,9 +71,9 @@ def find_liquidations_from_traces(
# Add the transfer # Add the transfer
liquidator = next( liquidator = next(
( (
contract_addresses[i] LIQUIDATION_CONTRACT_ADDRESSES[i]
for i in range(len(contract_addresses)) for i in range(len(LIQUIDATION_CONTRACT_ADDRESSES))
if trace.inputs["sender"] == contract_addresses[i] if trace.inputs["sender"] == LIQUIDATION_CONTRACT_ADDRESSES[i]
), ),
"", "",
) )
@ -87,9 +88,10 @@ def find_liquidations_from_traces(
# Add the transfer # Add the transfer
liquidator = next( liquidator = next(
( (
contract_addresses[i] LIQUIDATION_CONTRACT_ADDRESSES[i]
for i in range(len(contract_addresses)) for i in range(len(LIQUIDATION_CONTRACT_ADDRESSES))
if trace.inputs["recipient"] == contract_addresses[i] if trace.inputs["recipient"]
== LIQUIDATION_CONTRACT_ADDRESSES[i]
), ),
"", "",
) )
@ -108,7 +110,7 @@ def is_transfer_from_liquidator(
transfer = ERC20Transfer.from_trace(trace) transfer = ERC20Transfer.from_trace(trace)
if ( if (
trace.classification == Classification.transfer trace.classification == Classification.transfer
and transfer.from_address in contract_addresses and transfer.from_address in LIQUIDATION_CONTRACT_ADDRESSES
and trace.transaction_hash not in unique_transfer_hashes and trace.transaction_hash not in unique_transfer_hashes
): ):
return True return True
@ -121,7 +123,7 @@ def is_transfer_to_liquidator(trace: ClassifiedTrace) -> bool:
transfer = ERC20Transfer.from_trace(trace) transfer = ERC20Transfer.from_trace(trace)
if ( if (
trace.classification == Classification.transfer trace.classification == Classification.transfer
and transfer.to_address in contract_addresses and transfer.to_address in LIQUIDATION_CONTRACT_ADDRESSES
): ):
return True return True
else: else: