From ca0014533a19a35164a4db7f6a34bde49393241b Mon Sep 17 00:00:00 2001 From: Gui Heise Date: Thu, 18 Nov 2021 12:52:48 -0500 Subject: [PATCH] Add getter method for Uni recipient address --- mev_inspect/classifiers/specs/uniswap.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/mev_inspect/classifiers/specs/uniswap.py b/mev_inspect/classifiers/specs/uniswap.py index d597f80..017668b 100644 --- a/mev_inspect/classifiers/specs/uniswap.py +++ b/mev_inspect/classifiers/specs/uniswap.py @@ -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