Add declarator viewer cli

This commit is contained in:
nolash 2021-01-05 14:22:34 +01:00
parent a5f658d11b
commit 4311c4ee1d
Signed by: lash
GPG Key ID: 21D2E7BB88C2A746
2 changed files with 87 additions and 2 deletions

View File

@ -26,8 +26,8 @@ data_dir = os.path.join(script_dir, '..', 'data')
argparser = argparse.ArgumentParser()
argparser.add_argument('-p', '--provider', dest='p', default='http://localhost:8545', type=str, help='Web3 provider url (http only)')
argparser.add_argument('-r', '--contract-address', dest='r', type=str, help='Token endorsement contract address')
argparser.add_argument('-o', '--endorser-address', dest='o', type=str, help='Address to use to sign endorsement transaction')
argparser.add_argument('-r', '--contract-address', dest='r', type=str, help='Address declaration contract address')
argparser.add_argument('-o', '--declarator-address', dest='o', type=str, help='Signing address for the declaration')
argparser.add_argument('--abi-dir', dest='abi_dir', type=str, default=data_dir, help='Directory containing bytecode and abi (default: {})'.format(data_dir))
argparser.add_argument('-v', action='store_true', help='Be verbose')
argparser.add_argument('address', type=str, help='Ethereum address to add declaration to')

View File

@ -0,0 +1,85 @@
"""View address declaration, and attempt to resolve contents
.. moduleauthor:: Louis Holbrook <dev@holbrook.no>
.. pgp:: 0826EDA1702D1E87C6E2875121D2E7BB88C2A746
"""
# standard imports
import urllib
import os
import json
import argparse
import logging
# third-party imports
import web3
logging.basicConfig(level=logging.WARNING)
logg = logging.getLogger()
logging.getLogger('web3').setLevel(logging.WARNING)
logging.getLogger('urllib3').setLevel(logging.WARNING)
script_dir = os.path.dirname(__file__)
data_dir = os.path.join(script_dir, '..', 'data')
argparser = argparse.ArgumentParser()
argparser.add_argument('-p', '--provider', dest='p', default='http://localhost:8545', type=str, help='Web3 provider url (http only)')
argparser.add_argument('-r', '--contract-address', dest='r', type=str, help='Address declaration contract address')
argparser.add_argument('-o', '--declarator-address', dest='o', type=str, help='Signing address for the declaration')
argparser.add_argument('--resolve', action='store_true', help='Attempt to resolve the hashes to actual content')
argparser.add_argument('--resolve-http', dest='resolve_http', type=str, help='Base url to look up content hashes')
argparser.add_argument('--abi-dir', dest='abi_dir', type=str, default=data_dir, help='Directory containing bytecode and abi (default: {})'.format(data_dir))
argparser.add_argument('-v', action='store_true', help='Be verbose')
argparser.add_argument('address', type=str, help='Ethereum declaration address to look up')
args = argparser.parse_args()
if args.v:
logg.setLevel(logging.DEBUG)
def try_sha256(s):
r = urllib.request.urlopen(os.path.join(args.resolve_http, s.hex()))
return r.read().decode('utf-8')
def try_utf8(s):
return s.decode('utf-8')
def main():
w3 = web3.Web3(web3.Web3.HTTPProvider(args.p))
f = open(os.path.join(args.abi_dir, 'AddressDeclarator.json'), 'r')
abi = json.load(f)
f.close()
w3.eth.defaultAccount = w3.eth.accounts[0]
if args.o != None:
w3.eth.defaultAccount = args.o
logg.debug('endorser address {}'.format(w3.eth.defaultAccount))
c = w3.eth.contract(abi=abi, address=args.r)
declarations = c.functions.declaration(w3.eth.defaultAccount, args.address).call()
for d in declarations:
if not args.resolve:
print(d.hex())
continue
if args.resolve_http:
try:
r = try_sha256(d)
print(r)
continue
except urllib.error.HTTPError:
pass
try:
print(try_utf8(d))
except UnicodeDecodeError:
pass
if __name__ == '__main__':
main()