diff --git a/uniswap/uniswap4.py b/uniswap/uniswap4.py index 47ef25d..4e5ed4b 100644 --- a/uniswap/uniswap4.py +++ b/uniswap/uniswap4.py @@ -47,10 +47,10 @@ def __init__( self, address: Union[AddressLike, str, None], private_key: Optional[str], - provider: str = None, - web3: Web3 = None, + provider: Optional[str] = None, + web3: Optional[Web3] = None, default_slippage: float = 0.01, - poolmanager_contract_addr: str = None, + poolmanager_contract_addr: Optional[str] = None, ) -> None: """ :param address: The public address of the ETH wallet to use. @@ -91,7 +91,7 @@ def __init__( self.max_approval_check_int = int(max_approval_check_hex, 16) if poolmanager_contract_addr is None: - poolmanager_contract_addr = _poolmanager_contract_addresses[self.network] + self.poolmanager_contract_addr = _poolmanager_contract_addresses[self.network] self.router = _load_contract( self.w3, @@ -153,7 +153,7 @@ def get_price( ) try: - price = self.w3.eth.call(signed_txn) + price = int(self.w3.eth.call(signed_txn)) except ContractLogicError as revert: price = int(self.w3.codec.decode_abi(["int128[]","uint160","uint32"], revert.data)[1]) return price @@ -265,7 +265,7 @@ def currency_delta( """ :Get the current value of liquidity for the specified pool and position """ - currency_delta = self.router.functions.currencyDelta(locker, currency0).call() + currency_delta = int(self.router.functions.currencyDelta(locker, currency0).call()) return currency_delta def reserves_of( @@ -276,7 +276,7 @@ def reserves_of( :Get the current value in slot0 of the given pool """ - reserves = self.router.functions.reservesOf().call() + reserves = int(self.router.functions.reservesOf().call()) return reserves # ------ Pool manager WRITE methods ---------------------------------------------------------------- @@ -454,7 +454,7 @@ def modify_position( "params": modify_position_params, } ), - self._get_tx_params(value=qty), + self._get_tx_params(value=Wei(qty)), ) def settle( @@ -472,7 +472,7 @@ def settle( "currency ": currency0, } ), - self._get_tx_params(value=qty), + self._get_tx_params(value=Wei(qty)), ) def take( @@ -515,11 +515,7 @@ def get_token_balance(self, token: AddressLike) -> int: 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 = ( - self._exchange_address_from_token(token) - if self.version == 1 - else self.router_address - ) + contract_addr = poolmanager_contract_addr function = _load_contract_erc20(self.w3, token).functions.approve( contract_addr, max_approval ) @@ -595,23 +591,11 @@ def get_token(self, address: AddressLike, abi_name: str = "erc20") -> ERC20Token symbol = _symbol return ERC20Token(symbol, address, name, decimals) - def get_pool_id(self, currency0: AddressLike, currency1: AddressLike, fee : int, tickSpacing : int, hooks : AddressLike = ETH) -> bytes: + def get_pool_id(self, currency0: AddressLike, currency1: AddressLike, fee : int, tickSpacing : int, hooks : Union[AddressLike, str, None] = NOHOOK_ADDRESS) -> bytes: if int(currency0) > int(currency1): currency0 , currency1 = currency1 , currency0 - return self.w3.solidity_keccak(["address", "address", "int24", "int24", "address"], [(currency0, currency1, fee, tickSpacing, hooks)]) + pool_id = bytes(self.w3.solidity_keccak(["address", "address", "int24", "int24", "address"], [(currency0, currency1, fee, tickSpacing, hooks)])) + return - # ------ Test utilities ------------------------------------------------------------ - def _get_token_addresses(self) -> Dict[str, ChecksumAddress]: - """ - Returns a dict with addresses for tokens for the current net. - Used in testing. - """ - netid = int(self.w3.net.version) - netname = _netid_to_name[netid] - if netname == "mainnet": - return tokens - elif netname == "rinkeby": - return tokens_rinkeby - else: - raise Exception(f"Unknown net '{netname}'") + \ No newline at end of file pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy