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

Adding typing Union[] to _parse_jit_liquidity_instances and improving recursive spagetti

Browse Source
This commit is contained in:
elicb 2022-04-19 19:23:44 -07:00
parent a509f10cc9
commit 0a5518ff81
1 changed files with 16 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mev_inspect/jit_liquidity.py
View File

@ -1,4 +1,4 @@
from typing import List from typing import List, Union, Tuple
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
@ -42,7 +42,7 @@ def _parse_jit_liquidity_instance(
burn_trace: ClassifiedTrace, burn_trace: ClassifiedTrace,
classified_traces: List[ClassifiedTrace], classified_traces: List[ClassifiedTrace],
swaps: List[Swap], swaps: List[Swap],
) -> JITLiquidity: ) -> Union[JITLiquidity, None]:
valid_swaps = list(filter( valid_swaps = list(filter(
lambda t: mint_trace.transaction_position < t.transaction_position < burn_trace.transaction_position, lambda t: mint_trace.transaction_position < t.transaction_position < burn_trace.transaction_position,
swaps swaps
@ -95,18 +95,16 @@ def _parse_jit_liquidity_instance(
) )
def _get_token_order(token_a: str, token_b: str) -> (int, int): def _get_token_order(token_a: str, token_b: str) -> Tuple[str, str]:
token_order = True if int(token_a, 16) < int(token_b, 16) else False token_order = True if int(token_a, 16) < int(token_b, 16) else False
return (token_a, token_b) if token_order else (token_b, token_a) return (token_a, token_b) if token_order else (token_b, token_a)
def _get_bot_address( def _get_bot_address( # Janky and a half...
mint_trace: ClassifiedTrace, mint_trace: ClassifiedTrace,
classified_traces: List[ClassifiedTrace] classified_traces: List[ClassifiedTrace]
) -> str: ) -> str:
if mint_trace.from_address not in LIQUIDITY_MINT_ROUTERS: if mint_trace.from_address in LIQUIDITY_MINT_ROUTERS:
return mint_trace.from_address
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
@ -116,6 +114,5 @@ def _get_bot_address(
return _get_bot_address(bot_trace[0], classified_traces) return _get_bot_address(bot_trace[0], classified_traces)
else: else:
return "0x" + ("0" * 40) # get rid of this case by properly searching the trace_address return "0x" + ("0" * 40) # get rid of this case by properly searching the trace_address
_get_bot_address(bot_trace[0], classified_traces) return _get_bot_address(bot_trace[0], classified_traces)
return mint_trace.from_address