feat: add height param to balanceOf call

* defaults to 'latest'
This commit is contained in:
Mohamed Sohail 2022-11-21 15:49:34 +03:00 committed by lash
parent a87fbb13bc
commit 5e7e539bb9

View File

@ -9,12 +9,14 @@ 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,
TxFormat, TxFormat,
) )
from chainlib.jsonrpc import JSONRPCRequest from chainlib.jsonrpc import JSONRPCRequest
from chainlib.block import BlockSpec
from hexathon import ( from hexathon import (
add_0x, add_0x,
strip_0x, strip_0x,
@ -26,7 +28,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=BlockSpec.LATEST):
j = JSONRPCRequest(id_generator) j = JSONRPCRequest(id_generator)
o = j.template() o = j.template()
o['method'] = 'eth_call' o['method'] = 'eth_call'
@ -38,7 +40,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