52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
from typing import List
|
|
|
|
from mev_inspect.schemas.sandwiches import Sandwich
|
|
from mev_inspect.schemas.swaps import Swap
|
|
|
|
|
|
def get_sandwiches(swaps: List[Swap]) -> List[Sandwich]:
|
|
ordered_swaps = list(
|
|
sorted(
|
|
swaps,
|
|
key=lambda swap: (swap.transaction_position, swap.trace_address),
|
|
)
|
|
)
|
|
|
|
sandwiches: List[Sandwich] = []
|
|
|
|
for index, front_swap in enumerate(ordered_swaps):
|
|
sandwicher_address = front_swap.from_address
|
|
rest_swaps = ordered_swaps[index:]
|
|
|
|
sandwiched_swaps = []
|
|
|
|
for other_swap in rest_swaps:
|
|
if other_swap.transaction_hash == front_swap.transaction_hash:
|
|
continue
|
|
|
|
if other_swap.contract_address == front_swap.contract_address:
|
|
if (
|
|
other_swap.token_in_address == front_swap.token_in_address
|
|
and other_swap.token_out_address == front_swap.token_out_address
|
|
and other_swap.from_address != sandwicher_address
|
|
):
|
|
sandwiched_swaps.append(other_swap)
|
|
elif (
|
|
other_swap.token_out_address == front_swap.token_in_address
|
|
and other_swap.token_in_address == front_swap.token_out_address
|
|
and other_swap.from_address == sandwicher_address
|
|
):
|
|
sandwiches.append(
|
|
Sandwich(
|
|
block_number=front_swap.block_number,
|
|
sandwicher_address=sandwicher_address,
|
|
frontrun_transaction_hash=front_swap.transaction_hash,
|
|
backrun_transaction_hash=other_swap.transaction_hash,
|
|
sandwiched_swaps=sandwiched_swaps,
|
|
)
|
|
)
|
|
|
|
break
|
|
|
|
return sandwiches
|