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

Get uniswap v2 working more generally

Browse Source
This commit is contained in:
Luke Van Seters 2021-08-03 22:23:59 -04:00
parent 1b22a67616
commit 37d61a6614
1 changed files with 40 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
mev_inspect/strategies/arbitrage.py
View File

@ -106,7 +106,11 @@ def _parse_uniswap_v3_swap(
return None return None
pool_address = trace.to_address pool_address = trace.to_address
recipient_address = trace.inputs["recipient"] recipient_address = (
trace.inputs["recipient"]
if trace.inputs is not None and "recipient" in trace.inputs
else trace.from_address
)
child_transfers = _remove_inner_transfers( child_transfers = _remove_inner_transfers(
[ [
@ -152,13 +156,36 @@ def _parse_uniswap_v2_swap(
prior_traces: List[ClassifiedTrace], prior_traces: List[ClassifiedTrace],
child_traces: List[ClassifiedTrace], child_traces: List[ClassifiedTrace],
) -> Optional[Swap]: ) -> Optional[Swap]:
pool_address = trace.to_address pool_address = trace.to_address
transfers_to_pool = [ recipient_address = (
trace.inputs["to"]
if trace.inputs is not None and "to" in trace.inputs
else trace.from_address
)
prior_transfers = [
_as_transfer(prior_trace) _as_transfer(prior_trace)
for prior_trace in prior_traces for prior_trace in prior_traces
if prior_trace.classification == Classification.transfer
]
child_transfers = [
_as_transfer(child)
for child in child_traces
if child.classification == Classification.transfer
]
transfers_to_pool = [
transfer for transfer in prior_transfers if transfer.to_address == pool_address
]
transfers_from_pool_to_recipient = [
transfer
for transfer in child_transfers
if ( if (
prior_trace.classification == Classification.transfer transfer.to_address == recipient_address
and prior_trace.to_address == pool_address and transfer.from_address == pool_address
) )
] ]
@ -166,29 +193,23 @@ def _parse_uniswap_v2_swap(
if len(transfers_to_pool) == 0: if len(transfers_to_pool) == 0:
return None return None
most_recent_transfer_to_pool = transfers_to_pool[-1]
all_pool_internal_transfers = [
_as_transfer(child)
for child in child_traces
if child.classification == Classification.transfer
]
# expecting exactly one transfer inside the pool # expecting exactly one transfer inside the pool
if len(all_pool_internal_transfers) != 1: if len(transfers_from_pool_to_recipient) != 1:
return None return None
pool_internal_transfer = all_pool_internal_transfers[0] transfer_in = transfers_to_pool[-1]
transfer_out = transfers_from_pool_to_recipient[0]
return Swap( return Swap(
abi_name="UniswapV2Pair", abi_name="UniswapV2Pair",
transaction_hash=trace.transaction_hash, transaction_hash=trace.transaction_hash,
pool_address=pool_address, pool_address=pool_address,
from_address=most_recent_transfer_to_pool.from_address, from_address=transfer_in.from_address,
to_address=pool_internal_transfer.to_address, to_address=transfer_out.to_address,
token_in_address=most_recent_transfer_to_pool.token_address, token_in_address=transfer_in.token_address,
token_in_amount=most_recent_transfer_to_pool.amount, token_in_amount=transfer_in.amount,
token_out_address=pool_internal_transfer.token_address, token_out_address=transfer_out.token_address,
token_out_amount=pool_internal_transfer.amount, token_out_amount=transfer_out.amount,
) )