chainlib/cic_tools/eth/method.py
2021-02-09 09:42:46 +01:00

62 lines
1.4 KiB
Python

import sha3
import uuid
from hexathon import add_0x
from eth_abi import encode_single
from .hash import keccak256_string_to_hex
from .constant import ZERO_ADDRESS
# TODO: move to cic-contracts
erc20_balance_signature = keccak256_string_to_hex('balanceOf(address)')[:8]
erc20_decimals_signature = keccak256_string_to_hex('decimals()')[:8]
def jsonrpc_template():
return {
'jsonrpc': '2.0',
'id': str(uuid.uuid4()),
'method': None,
'params': [],
}
def erc20_balance(contract_address, address, sender_address=ZERO_ADDRESS):
o = jsonrpc_template()
o['method'] = 'eth_call'
data = erc20_balance_signature
data += encode_single('address', address).hex()
data = add_0x(data)
a = call(contract_address, data=data)
o['params'].append(a)
o['params'].append('latest')
return o
def erc20_decimals(contract_address, sender_address=ZERO_ADDRESS):
o = jsonrpc_template()
o['method'] = 'eth_call'
arg = add_0x(erc20_decimals_signature)
#o['params'].append(arg)
a = call(contract_address, arg)
o['params'].append(a)
o['params'].append('latest')
return o
def call(contract_address, data, sender_address=ZERO_ADDRESS):
return {
'from': sender_address,
'to': contract_address,
'data': data,
}
def jsonrpc_result(o, ep):
if o.get('error') != None:
raise ep.translate(o)
return o['result']