Nucypher
/
nucypher
mirror of https://github.com/nucypher/nucypher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Always name the middlewares we add to web3 middleware onion.

v7.4.1-hotfix^2
derekpierre 2024-08-28 11:52:37 -04:00
parent f692b0650f
commit 2ec5680a1f
No known key found for this signature in database
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
nucypher/blockchain/eth/clients.py
View File

@ -86,13 +86,13 @@ class EthereumClient:
endpoint_uri = getattr(self.w3.provider, "endpoint_uri", "")
if "infura" in endpoint_uri:
self.log.debug("Adding Infura RPC retry middleware to client")
self.add_middleware(InfuraRetryRequestMiddleware)
self.add_middleware(InfuraRetryRequestMiddleware, name="infura_retry")
elif "alchemyapi.io" in endpoint_uri:
self.log.debug("Adding Alchemy RPC retry middleware to client")
self.add_middleware(AlchemyRetryRequestMiddleware)
self.add_middleware(AlchemyRetryRequestMiddleware, name="alchemy_retry")
else:
self.log.debug("Adding RPC retry middleware to client")
self.add_middleware(RetryRequestMiddleware)
self.add_middleware(RetryRequestMiddleware, name="retry")
# poa middleware
chain_id = self.chain_id
@ -120,7 +120,7 @@ class EthereumClient:
# simple cache middleware
self.log.debug("Adding simple_cache_middleware")
self.add_middleware(simple_cache_middleware)
self.add_middleware(simple_cache_middleware, name="simple_cache")
@property
def chain_name(self) -> str:
@ -141,8 +141,8 @@ class EthereumClient:
def inject_middleware(self, middleware, **kwargs):
self.w3.middleware_onion.inject(middleware, **kwargs)
def add_middleware(self, middleware):
self.w3.middleware_onion.add(middleware)
def add_middleware(self, middleware, **kwargs):
self.w3.middleware_onion.add(middleware, **kwargs)
def set_gas_strategy(self, gas_strategy):
self.w3.eth.set_gas_price_strategy(gas_strategy)

2
tests/mock/interfaces.py
View File

@ -68,7 +68,7 @@ class MockEthereumClient(EthereumClient):
def __init__(self, w3):
super().__init__(w3=w3)
def add_middleware(self, middleware):
def add_middleware(self, middleware, **kwargs):
pass
def inject_middleware(self, middleware, **kwargs):