mirror of
git://holbrook.no/eth-contract-registry
synced 2024-12-22 12:27:31 +01:00
130 lines
3.5 KiB
Python
130 lines
3.5 KiB
Python
"""Set identifier value on contract registry
|
|
|
|
.. moduleauthor:: Louis Holbrook <dev@holbrook.no>
|
|
.. pgp:: 0826EDA1702D1E87C6E2875121D2E7BB88C2A746
|
|
|
|
"""
|
|
|
|
# standard imports
|
|
import sys
|
|
import os
|
|
import logging
|
|
|
|
# external imports
|
|
import chainlib.eth.cli
|
|
from chainlib.chain import ChainSpec
|
|
from chainlib.eth.tx import receipt
|
|
from chainlib.eth.constant import ZERO_CONTENT
|
|
from chainlib.eth.constant import ZERO_ADDRESS
|
|
from chainlib.error import JSONRPCException
|
|
from chainlib.eth.address import to_checksum_address
|
|
from hexathon import (
|
|
add_0x,
|
|
strip_0x,
|
|
)
|
|
from chainlib.eth.cli.arg import (
|
|
Arg,
|
|
ArgFlag,
|
|
process_args,
|
|
)
|
|
from chainlib.eth.cli.config import (
|
|
Config,
|
|
process_config,
|
|
)
|
|
from chainlib.eth.cli.log import process_log
|
|
from chainlib.eth.settings import process_settings
|
|
from chainlib.settings import ChainSettings
|
|
|
|
# local imports
|
|
from eth_contract_registry.registry import ContractRegistry
|
|
|
|
logging.basicConfig(level=logging.WARNING)
|
|
logg = logging.getLogger()
|
|
|
|
|
|
def process_config_local(config, arg, args, flags):
|
|
config.add(config.get('_POSARG'), '_IDENTIFIER')
|
|
return config
|
|
|
|
|
|
arg_flags = ArgFlag()
|
|
arg = Arg(arg_flags)
|
|
flags = arg_flags.STD_READ | arg_flags.EXEC
|
|
|
|
argparser = chainlib.eth.cli.ArgumentParser()
|
|
argparser = process_args(argparser, arg, flags)
|
|
argparser.add_argument('identifier', type=str, help='Contract identifier to look up')
|
|
args = argparser.parse_args()
|
|
|
|
logg = process_log(args, logg)
|
|
|
|
config = Config()
|
|
config = process_config(config, arg, args, flags, positional_name='identifier')
|
|
config = process_config_local(config, arg, args, flags)
|
|
logg.debug('config loaded:\n{}'.format(config))
|
|
|
|
settings = ChainSettings()
|
|
settings = process_settings(settings, config)
|
|
logg.debug('settings loaded:\n{}'.format(settings))
|
|
|
|
|
|
def out_element(e, w=sys.stdout):
|
|
if config.get('_RAW'):
|
|
w.write(e[1] + '\n')
|
|
else:
|
|
w.write(e[0] + '\t' + e[1] + '\n')
|
|
|
|
|
|
def element(ifc, conn, registry_address, identifier, w=sys.stdout, sender_address=ZERO_ADDRESS):
|
|
o = ifc.address_of(registry_address, identifier, sender_address=sender_address)
|
|
r = conn.do(o)
|
|
address = ifc.parse_address_of(r)
|
|
out_element((identifier, address), w)
|
|
|
|
|
|
def ls(ifc, conn, registry_address, w=sys.stdout, sender_address=ZERO_ADDRESS):
|
|
i = 0
|
|
while True:
|
|
o = ifc.identifier(registry_address, i, sender_address=sender_address)
|
|
try:
|
|
r = conn.do(o)
|
|
identifier = ifc.parse_identifier(r)
|
|
element(ifc, conn, registry_address, identifier, w)
|
|
i += 1
|
|
except JSONRPCException:
|
|
break
|
|
|
|
|
|
def main():
|
|
c = ContractRegistry(
|
|
settings.get('CHAIN_SPEC')
|
|
)
|
|
|
|
identifier = config.get('_IDENTIFIER')
|
|
|
|
registry_address = to_checksum_address(config.get('_EXEC_ADDRESS'))
|
|
if not config.true('_UNSAFE') and strip_0x(registry_address) != strip_0x(config.get('_EXEC_ADDRESS')):
|
|
raise ValueError('invalid checksum address for contract')
|
|
|
|
if identifier != None:
|
|
element(
|
|
c,
|
|
settings.get('CONN'),
|
|
settings.get('EXEC'),
|
|
config.get('_IDENTIFIER'),
|
|
w=sys.stdout,
|
|
sender_address=settings.get('SENDER_ADDRESS'),
|
|
)
|
|
else:
|
|
ls(
|
|
c,
|
|
settings.get('CONN'),
|
|
settings.get('EXEC'),
|
|
w=sys.stdout,
|
|
sender_address=settings.get('SENDER_ADDRESS'),
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|