From 784922fa07cc9f23b3dfbd348eca4fc069b4e180 Mon Sep 17 00:00:00 2001 From: Gui Heise Date: Mon, 22 Nov 2021 12:07:30 -0500 Subject: [PATCH 1/2] Rename to helpers, add func --- mev_inspect/classifiers/{swaps.py => helpers.py} | 11 +++++++++++ mev_inspect/classifiers/specs/balancer.py | 2 +- mev_inspect/classifiers/specs/curve.py | 2 +- mev_inspect/classifiers/specs/uniswap.py | 2 +- 4 files changed, 14 insertions(+), 3 deletions(-) rename mev_inspect/classifiers/{swaps.py => helpers.py} (90%) diff --git a/mev_inspect/classifiers/swaps.py b/mev_inspect/classifiers/helpers.py similarity index 90% rename from mev_inspect/classifiers/swaps.py rename to mev_inspect/classifiers/helpers.py index 58d6440..b1f9d2d 100644 --- a/mev_inspect/classifiers/swaps.py +++ b/mev_inspect/classifiers/helpers.py @@ -84,3 +84,14 @@ def _filter_transfers( filtered_transfers.append(transfer) return filtered_transfers + + +def get_amount_transferred_to_address( + address: str, transfers: List[Transfer] +) -> Optional[int]: + + for transfer in transfers: + if transfer.to_address == address: + return transfer.amount + + raise ValueError(f"Transfer to {address} not found.") diff --git a/mev_inspect/classifiers/specs/balancer.py b/mev_inspect/classifiers/specs/balancer.py index 2614002..89ae1f1 100644 --- a/mev_inspect/classifiers/specs/balancer.py +++ b/mev_inspect/classifiers/specs/balancer.py @@ -9,7 +9,7 @@ from mev_inspect.schemas.classifiers import ( ClassifierSpec, SwapClassifier, ) -from mev_inspect.classifiers.swaps import create_swap_from_transfers +from mev_inspect.classifiers.helpers import create_swap_from_transfers BALANCER_V1_POOL_ABI_NAME = "BPool" diff --git a/mev_inspect/classifiers/specs/curve.py b/mev_inspect/classifiers/specs/curve.py index 08244ea..2762725 100644 --- a/mev_inspect/classifiers/specs/curve.py +++ b/mev_inspect/classifiers/specs/curve.py @@ -10,7 +10,7 @@ from mev_inspect.schemas.classifiers import ( ClassifierSpec, SwapClassifier, ) -from mev_inspect.classifiers.swaps import create_swap_from_transfers +from mev_inspect.classifiers.helpers import create_swap_from_transfers class CurveSwapClassifier(SwapClassifier): diff --git a/mev_inspect/classifiers/specs/uniswap.py b/mev_inspect/classifiers/specs/uniswap.py index 017668b..ce8b634 100644 --- a/mev_inspect/classifiers/specs/uniswap.py +++ b/mev_inspect/classifiers/specs/uniswap.py @@ -9,7 +9,7 @@ from mev_inspect.schemas.classifiers import ( ClassifierSpec, SwapClassifier, ) -from mev_inspect.classifiers.swaps import create_swap_from_transfers +from mev_inspect.classifiers.helpers import create_swap_from_transfers UNISWAP_V2_PAIR_ABI_NAME = "UniswapV2Pair" From 89c2ed3a84768b8a2a039f3110be89167a79414f Mon Sep 17 00:00:00 2001 From: Gui Heise Date: Mon, 22 Nov 2021 12:16:39 -0500 Subject: [PATCH 2/2] Remove func --- mev_inspect/classifiers/helpers.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/mev_inspect/classifiers/helpers.py b/mev_inspect/classifiers/helpers.py index b1f9d2d..58d6440 100644 --- a/mev_inspect/classifiers/helpers.py +++ b/mev_inspect/classifiers/helpers.py @@ -84,14 +84,3 @@ def _filter_transfers( filtered_transfers.append(transfer) return filtered_transfers - - -def get_amount_transferred_to_address( - address: str, transfers: List[Transfer] -) -> Optional[int]: - - for transfer in transfers: - if transfer.to_address == address: - return transfer.amount - - raise ValueError(f"Transfer to {address} not found.")