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

Finalizing Docstrings and modifying to use search dict instead of list

Browse Source
This commit is contained in:
elicb 2022-05-02 11:25:37 -07:00
parent 293c59d7c6
commit 6701ff4cad
1 changed files with 32 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
mev_inspect/transfers.py
View File

@ -132,16 +132,16 @@ def get_net_transfers(
classified_traces: List[ClassifiedTrace], classified_traces: List[ClassifiedTrace],
) -> List[Transfer]: ) -> List[Transfer]:
""" """
Super Jank...
Returns the net transfers per transaction from a list of Classified Traces. Returns the net transfers per transaction from a list of Classified Traces.
Ex. if a bot transfers 200 WETH to a contract, and the contract transfers the excess 50 WETH back to the bot, If A transfers 200WETH to B ,and later in the transaction, B transfers 50WETH to A,
the following transfer would be returned (from_address=bot, to_address=contract, amount=150) the following transfer would be returned (from_address=A, to_address=B, amount=150)
if the contract transferred 300 WETH back to the bot, the following would be returned
(from_address=contract, to_address=bot, amount=100). if the contract transferred back 200 WETH, If B transferred 300WETH to A, the following would be returned
no transfer would be returned. (from_address=contract, to_address=bot, amount=100)
Additionally, ignores transfers forwarded from proxy contracts & uses initial proxy address
If B transferred 200WETH to A, no transfer would be returned
@param classified_traces: @param classified_traces:
@return: List of Transfer objects representing the net movement from A to B @return: List of Transfer objects representing the net movement of tokens
""" """
found_transfers: List[list] = [] found_transfers: List[list] = []
return_transfers: List[Transfer] = [] return_transfers: List[Transfer] = []
@ -156,36 +156,39 @@ def get_net_transfers(
continue continue
if trace.function_signature == "transfer(address,uint256)": if trace.function_signature == "transfer(address,uint256)":
net_search_info = [ net_search_info = {
trace.inputs["recipient"], "to_address": trace.inputs["recipient"],
trace.to_address, "token_address": trace.to_address,
trace.from_address, "from_address": trace.from_address,
] }
else: # trace.function_signature == "transferFrom(address,address,uint256)" elif trace.function_signature == "transferFrom(address,address,uint256)":
net_search_info = [ net_search_info = {
trace.inputs["recipient"], "to_address": trace.inputs["recipient"],
trace.to_address, "token_address": trace.to_address,
trace.inputs["sender"], "from_address": trace.inputs["sender"],
] }
if sorted(net_search_info) in found_transfers: else:
continue
if sorted(list(net_search_info.values())) in found_transfers:
for index, transfer in enumerate(return_transfers): for index, transfer in enumerate(return_transfers):
if ( if (
transfer.token_address != net_search_info[1] transfer.token_address != net_search_info["token_address"]
or transfer.transaction_hash != trace.transaction_hash or transfer.transaction_hash != trace.transaction_hash
): ):
continue continue
if ( if (
transfer.from_address == net_search_info[2] transfer.from_address == net_search_info["from_address"]
and transfer.to_address == net_search_info[0] and transfer.to_address == net_search_info["to_address"]
): ):
return_transfers[index].amount += trace.inputs["amount"] return_transfers[index].amount += trace.inputs["amount"]
return_transfers[index].trace_address = [-1] return_transfers[index].trace_address = [-1]
if ( if (
transfer.from_address == net_search_info[0] transfer.from_address == net_search_info["to_address"]
and transfer.to_address == net_search_info[2] and transfer.to_address == net_search_info["from_address"]
): ):
return_transfers[index].amount -= trace.inputs["amount"] return_transfers[index].amount -= trace.inputs["amount"]
return_transfers[index].trace_address = [-1] return_transfers[index].trace_address = [-1]
@ -196,13 +199,13 @@ def get_net_transfers(
block_number=trace.block_number, block_number=trace.block_number,
transaction_hash=trace.transaction_hash, transaction_hash=trace.transaction_hash,
trace_address=trace.trace_address, trace_address=trace.trace_address,
from_address=net_search_info[2], # Janky... improve from_address=net_search_info["from_address"],
to_address=net_search_info[0], to_address=net_search_info["to_address"],
amount=trace.inputs["amount"], amount=trace.inputs["amount"],
token_address=net_search_info[1], token_address=net_search_info["token_address"],
) )
) )
found_transfers.append(sorted(net_search_info)) found_transfers.append(sorted(list(net_search_info.values())))
process_index = -1 process_index = -1
while True: while True: