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

Add getter method for Uni recipient address

Browse Source
This commit is contained in:
Gui Heise 2021-11-18 12:52:48 -05:00
parent c5621e0676
commit ca0014533a
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mev_inspect/classifiers/specs/uniswap.py
View File

@ -24,10 +24,7 @@ class UniswapV3SwapClassifier(SwapClassifier):
child_transfers: List[Transfer], child_transfers: List[Transfer],
) -> Optional[Swap]: ) -> Optional[Swap]:
if "recipient" in trace.inputs: recipient_address = trace.inputs.get("recipient", trace.from_address)
recipient_address = trace.inputs["recipient"]
else:
recipient_address = trace.from_address
swap = create_swap_from_transfers( swap = create_swap_from_transfers(
trace, recipient_address, prior_transfers, child_transfers trace, recipient_address, prior_transfers, child_transfers
@ -43,10 +40,7 @@ class UniswapV2SwapClassifier(SwapClassifier):
child_transfers: List[Transfer], child_transfers: List[Transfer],
) -> Optional[Swap]: ) -> Optional[Swap]:
if "to" in trace.inputs: recipient_address = trace.inputs.get("to", trace.from_address)
recipient_address = trace.inputs["to"]
else:
recipient_address = trace.from_address
swap = create_swap_from_transfers( swap = create_swap_from_transfers(
trace, recipient_address, prior_transfers, child_transfers trace, recipient_address, prior_transfers, child_transfers