mev-inspect-py/liquidation_inspector.py
Guilherme Peyrelongue Heise 508847b8df TODO
2021-08-06 02:21:46 -04:00

114 lines
3.4 KiB
Python

# Internal imports
from mev_inspect import utils
from mev_inspect.config import load_config
from mev_inspect.schemas.blocks import NestedTrace, TraceType
from mev_inspect.schemas.classified_traces import Classification, ClassifiedTrace
from mev_inspect.schemas.strategy import StrategyType, Strategy, Liquidation
from mev_inspect.classifier_specs import CLASSIFIER_SPECS
from mev_inspect.trace_classifier import TraceClassifier
from mev_inspect import block
# External Libraries
import json
from typing import Optional
from web3 import Web3
from typing import List, Optional
# TODO: adjust profit to new model
# unit test
# coinbase check / collateral source
#
#
# Block inspect
# poetry run inspect -b 12498502 -r 'http://162.55.96.141:8546/'
#
#
#
all_traces = []
result = []
addrs = ['0x3dfd23A6c5E8BbcFc9581d2E864a68feb6a076d3']
transfers = []
# Inspect list of classified traces and identify liquidation
def liquidations(traces: List[ClassifiedTrace]):
event = []
# For each trace
for k in range(len(traces)):
trace = traces[k]
# Check for liquidation and register trace and unique liquidator
if trace.classification == Classification.liquidate:
all_traces.append(trace)
liquidator = trace.from_address
if liquidator not in addrs: addrs.append(liquidator)
print(f"\nLiquidation found: \n\t{liquidator}")
print(f"\nHash: \n\t{trace.transaction_hash}\n")
# Found liquidation, now parse inputs for data
for i in trace.inputs:
if(i == '_purchaseAmount'):
liquidation_amount = trace.inputs[i]
elif (i == '_collateral'):
collateral_type = trace.inputs[i]
elif (i == '_reserve'):
reserve = trace.inputs[i]
elif(i == '_user'):
liquidated_usr = trace.inputs[i]
# Register liquidation
result.append(
Liquidation(collateral_type=collateral_type,
collateral_amount=liquidation_amount,
collateral_source="",
reserve=reserve,
strategy=StrategyType.liquidation,
traces=all_traces,
protocols=[trace.protocol],)
)
# Check for transfer from a registered liquidator
elif (
trace.classification == Classification.transfer and
'sender' in trace.inputs and
trace.inputs['sender'] in addrs
):
liquidator = next(addrs[i] for i in range(len(addrs)) if trace.inputs['sender'] == addrs[i])
transfers.append(trace)
print(f"""
\nTransfer from liquidator {liquidator}:
\n\tAmount in received token: {trace.inputs['amount']} to
\n\t{trace.inputs['recipient']}
\n\tTransaction: {trace.transaction_hash}\n
""")
# Check for transfer to a registered liquidator
elif (
trace.classification == Classification.transfer and
trace.inputs['recipient'] in addrs
):
liquidator = next(addrs[i] for i in range(len(addrs)) if trace.inputs['recipient'] == addrs[i])
transfers.append(trace)
print(f"""
\nTransfer to liquidator {liquidator}:
\n\tAmount in received token: {trace.inputs['amount']} from
\n\t{trace.from_address}
\n\tTransaction: {trace.transaction_hash}\n
""")
return result
rpc = 'http://162.55.96.141:8546/'
block_number = 12498502
base_provider = Web3.HTTPProvider(rpc)
block_data = block.create_from_block_number(block_number, base_provider)
trace_clasifier = TraceClassifier(CLASSIFIER_SPECS)
classified_traces = trace_clasifier.classify(block_data.traces)
fin = liquidations(classified_traces)