feat: add height param to balanceOf call

* defaults to 'latest'
This commit is contained in:
Mohamed Sohail 2022-11-18 16:51:24 +03:00
parent 14ecb75b4e
commit fe0dcf1765
Signed by: kamikazechaser
GPG Key ID: 7DD45520C01CD85D

View File

@ -9,6 +9,7 @@ from chainlib.eth.contract import (
ABIContractType, ABIContractType,
abi_decode_single, abi_decode_single,
) )
from chainlib.eth.jsonrpc import to_blockheight_param
from chainlib.eth.error import RequestMismatchException from chainlib.eth.error import RequestMismatchException
from chainlib.eth.tx import ( from chainlib.eth.tx import (
TxFactory, TxFactory,
@ -26,7 +27,7 @@ logg = logging.getLogger()
class ERC20(TxFactory): class ERC20(TxFactory):
def balance_of(self, contract_address, address, sender_address=ZERO_ADDRESS, id_generator=None): def balance_of(self, contract_address, address, sender_address=ZERO_ADDRESS, id_generator=None, height=None):
j = JSONRPCRequest(id_generator) j = JSONRPCRequest(id_generator)
o = j.template() o = j.template()
o['method'] = 'eth_call' o['method'] = 'eth_call'
@ -38,7 +39,8 @@ class ERC20(TxFactory):
tx = self.template(sender_address, contract_address) tx = self.template(sender_address, contract_address)
tx = self.set_code(tx, data) tx = self.set_code(tx, data)
o['params'].append(self.normalize(tx)) o['params'].append(self.normalize(tx))
o['params'].append('latest') height = to_blockheight_param(height)
o['params'].append(height)
o = j.finalize(o) o = j.finalize(o)
return o return o