Skip to content

Commit 9731d89

Browse files
chore: black .
1 parent 555c309 commit 9731d89

File tree

2 files changed

+28
-23
lines changed

2 files changed

+28
-23
lines changed

multicall/multicall.py

Lines changed: 26 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -111,47 +111,51 @@ async def coroutine(self) -> Dict[str, Any]:
111111

112112
def _contract_method(self, request_signature: list, return_signature: tuple) -> None:
113113
self.calls.append(Call(self.multicall_address, request_signature, [return_signature]))
114-
115-
def add_base_fee(self, return_signature: tuple = ('base_fee', None)) -> None:
116-
signature = ['getBasefee()(uint256)']
114+
115+
def add_base_fee(self, return_signature: tuple = ("base_fee", None)) -> None:
116+
signature = ["getBasefee()(uint256)"]
117117
self._contract_method(signature, return_signature)
118118

119-
def add_block_hash(self, block_number: int, return_signature: tuple = ('block_hash', None)) -> None:
120-
signature = ['getBlockHash(uint256)(bytes32)', block_number]
119+
def add_block_hash(
120+
self, block_number: int, return_signature: tuple = ("block_hash", None)
121+
) -> None:
122+
signature = ["getBlockHash(uint256)(bytes32)", block_number]
121123
self._contract_method(signature, return_signature)
122124

123-
def add_block_number(self, return_signature: tuple = ('block_number', None)) -> None:
124-
signature = ['getBlockNumber()(uint256)']
125+
def add_block_number(self, return_signature: tuple = ("block_number", None)) -> None:
126+
signature = ["getBlockNumber()(uint256)"]
125127
self._contract_method(signature, return_signature)
126128

127-
def add_chain_id(self, return_signature: tuple = ('chain_id', None)) -> None:
128-
signature = ['getChainId()(uint256)']
129+
def add_chain_id(self, return_signature: tuple = ("chain_id", None)) -> None:
130+
signature = ["getChainId()(uint256)"]
129131
self._contract_method(signature, return_signature)
130132

131-
def add_block_coinbase(self, return_signature: tuple = ('coinbase', None)) -> None:
132-
signature = ['getCurrentBlockCoinbase()(address)']
133+
def add_block_coinbase(self, return_signature: tuple = ("coinbase", None)) -> None:
134+
signature = ["getCurrentBlockCoinbase()(address)"]
133135
self._contract_method(signature, return_signature)
134136

135-
def add_block_difficulty(self, return_signature: tuple = ('difficulty', None)) -> None:
136-
signature = ['getCurrentBlockDifficulty()(address)']
137+
def add_block_difficulty(self, return_signature: tuple = ("difficulty", None)) -> None:
138+
signature = ["getCurrentBlockDifficulty()(address)"]
137139
self._contract_method(signature, return_signature)
138140

139-
def add_block_gas_limit(self, return_signature: tuple = ('gas_limit', None)) -> None:
140-
signature = ['getCurrentBlockGasLimit()(uint256)']
141+
def add_block_gas_limit(self, return_signature: tuple = ("gas_limit", None)) -> None:
142+
signature = ["getCurrentBlockGasLimit()(uint256)"]
141143
self._contract_method(signature, return_signature)
142144

143-
def add_block_timestamp(self, return_signature: tuple = ('timestamp', None)) -> None:
144-
signature = ['getCurrentBlockTimestamp()(uint256)']
145+
def add_block_timestamp(self, return_signature: tuple = ("timestamp", None)) -> None:
146+
signature = ["getCurrentBlockTimestamp()(uint256)"]
145147
self._contract_method(signature, return_signature)
146148

147-
def add_eth_balance(self, address: AnyAddress, return_signature: tuple = ('eth_balance', None)) -> None:
148-
signature = ['getEthBalance(address)(uint256)', address]
149+
def add_eth_balance(
150+
self, address: AnyAddress, return_signature: tuple = ("eth_balance", None)
151+
) -> None:
152+
signature = ["getEthBalance(address)(uint256)", address]
149153
self._contract_method(signature, return_signature)
150154

151-
def add_last_block_hash(self, return_signature: tuple = ('last_block_hash', None)) -> None:
152-
signature = ['getLastBlockHash()(bytes32)']
155+
def add_last_block_hash(self, return_signature: tuple = ("last_block_hash", None)) -> None:
156+
signature = ["getLastBlockHash()(bytes32)"]
153157
self._contract_method(signature, return_signature)
154-
158+
155159
async def fetch_outputs(
156160
self, calls: List[Call], ConnErr_retries: int = 0, id: str = ""
157161
) -> List[CallResponse]:

tests/test_multicall.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,7 @@ def test_batcher_split_calls_odd():
160160
assert len(split[0]) == 14_999
161161
assert len(split[1]) == 15_000
162162

163+
163164
def test_multicalll_contract_methods():
164165
calls = [Call(CHAI, ["totalSupply()(uint)", CHAI, 1], [["totalSupply", None]])]
165166
multi = Multicall(calls)
@@ -171,7 +172,7 @@ def test_multicalll_contract_methods():
171172
multi.add_block_difficulty()
172173
multi.add_block_gas_limit()
173174
multi.add_block_timestamp()
174-
multi.add_eth_balance(CHAI, ('custom_balance', None))
175+
multi.add_eth_balance(CHAI, ("custom_balance", None))
175176
multi.add_last_block_hash()
176177

177178
result = multi()

0 commit comments

Comments
 (0)