diff --git a/uniswap/decorators.py b/uniswap/decorators.py index c5bbf0e..476bc85 100644 --- a/uniswap/decorators.py +++ b/uniswap/decorators.py @@ -26,12 +26,7 @@ def approved(self: Any, *args: Any, **kwargs: Any) -> Any: is_approved = self._is_approved(token) # logger.warning(f"Approved? {token}: {is_approved}") if not is_approved: - self._approve(token) - if token_two: - is_approved = self._is_approved(token_two) - # logger.warning(f"Approved? {token_two}: {is_approved}") - if not is_approved: - self._approve(token_two) + self.approve(token) return method(self, *args, **kwargs) return approved diff --git a/uniswap/uniswap.py b/uniswap/uniswap.py index 0a6e0c4..4ff6885 100644 --- a/uniswap/uniswap.py +++ b/uniswap/uniswap.py @@ -871,7 +871,7 @@ def remove_liquidity(self, token: str, max_token: int) -> HexBytes: return self._build_and_send_tx(function) # ------ Approval Utils ------------------------------------------------------------ - def _approve(self, token: AddressLike, max_approval: Optional[int] = None) -> None: + def approve(self, token: AddressLike, max_approval: Optional[int] = None) -> None: """Give an exchange/router max approval of a token.""" max_approval = self.max_approval_int if not max_approval else max_approval contract_addr = (