Add getter method for Uni recipient address

This commit is contained in:
Gui Heise 2021-11-18 12:52:48 -05:00
parent c5621e0676
commit ca0014533a

View File

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