mmgen/mmgen-wallet
1
Fork 0
Code Releases 28 Wiki Activity

proto.eth.contract: relocate methods

Browse source
This commit is contained in:
The MMGen Project 2025-03-20 08:48:03 +03:00
commit b6c285c02a
Signed by: mmgen
GPG key ID: 3F8B1861E32B7DA2
1 changed files with 48 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

97
mmgen/proto/eth/contract.py
View file

@ -36,6 +36,9 @@ def parse_abi(s):
class Contract:
def strip(self, s):
return ''.join([chr(b) for b in s if 32 <= b <= 127]).strip()
def create_method_id(self, sig):
return self.keccak_256(sig.encode()).hexdigest()[:8]
@ -47,6 +50,9 @@ class Contract:
int(parse_abi(data)[-1], 16) * self.base_unit,
from_decimal = True)
async def code(self):
return (await self.rpc.call('eth_getCode', '0x'+self.addr))[2:]
async def do_call(self, method_sig, method_args='', *, toUnit=False):
data = self.create_method_id(method_sig) + method_args
if self.cfg.debug:
@ -60,14 +66,52 @@ class Contract:
else:
return ret
async def txsign(self, tx_in, key, from_addr, *, chain_id=None):
from .pyethereum.transactions import Transaction
if chain_id is None:
res = await self.rpc.call('eth_chainId')
chain_id = None if res is None else int(res, 16)
etx = Transaction(**tx_in).sign(key, chain_id)
if etx.sender.hex() != from_addr:
die(3, f'Sender address {from_addr!r} does not match address of key {etx.sender.hex()!r}!')
if self.cfg.debug:
msg('TOKEN DATA:')
pp_msg(etx.to_dict())
msg('PARSED ABI DATA:\n {}'.format(
'\n '.join(parse_abi(etx.data.hex()))))
return namedtuple('signed_contract_transaction', ['etx', 'txhex', 'txid'])(
etx,
rlp.encode(etx).hex(),
CoinTxID(etx.hash.hex()))
async def txsend(self, txhex):
return (await self.rpc.call('eth_sendRawTransaction', '0x'+txhex)).replace('0x', '', 1)
class Token(Contract):
def __init__(self, cfg, proto, addr, decimals, *, rpc=None):
if type(self).__name__ == 'Token':
from ...util2 import get_keccak
self.keccak_256 = get_keccak(cfg)
self.cfg = cfg
self.proto = proto
self.addr = TokenAddr(proto, addr)
assert isinstance(decimals, int), f'decimals param must be int instance, not {type(decimals)}'
self.decimals = decimals
self.base_unit = Decimal('10') ** -self.decimals
self.rpc = rpc
async def get_balance(self, acct_addr):
return self.proto.coin_amt(
await self.do_call('balanceOf(address)', acct_addr.rjust(64, '0'), toUnit=True),
from_decimal = True)
def strip(self, s):
return ''.join([chr(b) for b in s if 32 <= b <= 127]).strip()
async def get_name(self):
return self.strip(bytes.fromhex((await self.do_call('name()'))[2:]))
@ -94,9 +138,6 @@ class Contract:
'decimals:', self.decimals,
'total supply:', await self.get_total_supply())
async def code(self):
return (await self.rpc.call('eth_getCode', '0x'+self.addr))[2:]
def create_data(
self,
to_addr,
@ -129,35 +170,7 @@ class Contract:
'nonce': nonce,
'data': bytes.fromhex(data)}
async def txsign(self, tx_in, key, from_addr, *, chain_id=None):
from .pyethereum.transactions import Transaction
if chain_id is None:
res = await self.rpc.call('eth_chainId')
chain_id = None if res is None else int(res, 16)
etx = Transaction(**tx_in).sign(key, chain_id)
if etx.sender.hex() != from_addr:
die(3, f'Sender address {from_addr!r} does not match address of key {etx.sender.hex()!r}!')
if self.cfg.debug:
msg('TOKEN DATA:')
pp_msg(etx.to_dict())
msg('PARSED ABI DATA:\n {}'.format(
'\n '.join(parse_abi(etx.data.hex()))))
return namedtuple('signed_contract_transaction', ['etx', 'txhex', 'txid'])(
etx,
rlp.encode(etx).hex(),
CoinTxID(etx.hash.hex()))
# The following are used for token deployment only:
async def txsend(self, txhex):
return (await self.rpc.call('eth_sendRawTransaction', '0x'+txhex)).replace('0x', '', 1)
# used for token deployment only:
async def transfer(
self,
from_addr,
@ -178,20 +191,6 @@ class Contract:
res = await self.txsign(tx_in, key, from_addr)
return await self.txsend(res.txhex)
class Token(Contract):
def __init__(self, cfg, proto, addr, decimals, *, rpc=None):
if type(self).__name__ == 'Token':
from ...util2 import get_keccak
self.keccak_256 = get_keccak(cfg)
self.cfg = cfg
self.proto = proto
self.addr = TokenAddr(proto, addr)
assert isinstance(decimals, int), f'decimals param must be int instance, not {type(decimals)}'
self.decimals = decimals
self.base_unit = Decimal('10') ** -self.decimals
self.rpc = rpc
class ResolvedToken(Token, metaclass=AsyncInit):
async def __init__(self, cfg, proto, rpc, addr):