crowetic/mev-inspect-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Clarifying logic of recursive _get_bot_address

Browse Source
This commit is contained in:
elicb 2022-04-19 19:44:44 -07:00
parent 839bcedce4
commit 4f03d107ce
1 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
mev_inspect/jit_liquidity.py
View File

@ -1,4 +1,4 @@
from typing import List, Union, Tuple from typing import List, Union, Tuple, Optional
from mev_inspect.schemas.jit_liquidity import JITLiquidity from mev_inspect.schemas.jit_liquidity import JITLiquidity
from mev_inspect.schemas.swaps import Swap from mev_inspect.schemas.swaps import Swap
@ -78,6 +78,8 @@ def _parse_jit_liquidity_instance(
token0_swap_volume += swap.token_in_amount if swap.token_in_address == token0_address else 0 token0_swap_volume += swap.token_in_amount if swap.token_in_address == token0_address else 0
token1_swap_volume += 0 if swap.token_in_address == token0_address else swap.token_in_amount token1_swap_volume += 0 if swap.token_in_address == token0_address else swap.token_in_amount
token_order = mint_transfers[0].token_address == token0_address
return JITLiquidity( return JITLiquidity(
block_number=mint_trace.block_number, block_number=mint_trace.block_number,
bot_address=bot_address, bot_address=bot_address,
@ -89,10 +91,10 @@ def _parse_jit_liquidity_instance(
swaps=jit_swaps, swaps=jit_swaps,
token0_address=token0_address, token0_address=token0_address,
token1_address=token1_address, token1_address=token1_address,
mint_token0_amount=mint_transfers[0].amount if mint_transfers[0].token_address == token0_address else mint_transfers[1].amount, mint_token0_amount=mint_transfers[0].amount if token_order else mint_transfers[1].amount,
mint_token1_amount=mint_transfers[1].amount if mint_transfers[0].token_address == token0_address else mint_transfers[0].amount, mint_token1_amount=mint_transfers[1].amount if token_order else mint_transfers[0].amount,
burn_token0_amount=burn_transfers[0].amount if burn_transfers[0].token_address == token0_address else burn_transfers[1].amount, burn_token0_amount=burn_transfers[0].amount if token_order else burn_transfers[1].amount,
burn_token1_amount=burn_transfers[1].amount if burn_transfers[0].token_address == token0_address else burn_transfers[0].amount, burn_token1_amount=burn_transfers[1].amount if token_order else burn_transfers[0].amount,
token0_swap_volume=token0_swap_volume, token0_swap_volume=token0_swap_volume,
token1_swap_volume=token1_swap_volume, token1_swap_volume=token1_swap_volume,
) )
@ -106,16 +108,17 @@ def _get_token_order(token_a: str, token_b: str) -> Tuple[str, str]:
def _get_bot_address( # Janky and a half... def _get_bot_address( # Janky and a half...
mint_trace: ClassifiedTrace, mint_trace: ClassifiedTrace,
classified_traces: List[ClassifiedTrace] classified_traces: List[ClassifiedTrace]
) -> str: ) -> Union[str, None]:
if mint_trace.from_address in LIQUIDITY_MINT_ROUTERS: if mint_trace.from_address in LIQUIDITY_MINT_ROUTERS:
bot_trace = list(filter( bot_trace = list(filter(
lambda t: t.to_address == mint_trace.from_address and t.transaction_hash == mint_trace.transaction_hash, lambda t: t.to_address == mint_trace.from_address and t.transaction_hash == mint_trace.transaction_hash,
classified_traces classified_traces
)) ))
if len(bot_trace) > 1: if len(bot_trace) == 1:
if is_child_trace_address(bot_trace[1].trace_address, bot_trace[0].trace_address): return _get_bot_address(bot_trace[0], classified_traces)
return _get_bot_address(bot_trace[0], classified_traces) elif is_child_trace_address(bot_trace[1].trace_address, bot_trace[0].trace_address):
else: return _get_bot_address(bot_trace[0], classified_traces)
return "0x" + ("0" * 40) # get rid of this case by properly searching the trace_address else:
return _get_bot_address(bot_trace[0], classified_traces) return None
return mint_trace.from_address return mint_trace.from_address