67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
"""Adds writers and/or account entries to accounts index
|
||
|
||
.. moduleauthor:: Louis Holbrook <dev@holbrook.no>
|
||
.. pgp:: 0826EDA1702D1E87C6E2875121D2E7BB88C2A746
|
||
|
||
"""
|
||
|
||
# standard imports
|
||
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('-w', '--writer', dest='w', action='append', type=str, help='Writer to add')
|
||
argparser.add_argument('-o', '--owner', dest='o', type=str, help='Accounts index owner')
|
||
argparser.add_argument('-a', '--account', dest='a', action='append', type=str, help='Account to add')
|
||
argparser.add_argument('-r', '--accounts-index-address', dest='r', required=True, type=str, help='Contract address to account index to edit')
|
||
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')
|
||
args = argparser.parse_args()
|
||
|
||
if args.v:
|
||
logg.setLevel(logging.DEBUG)
|
||
|
||
def main():
|
||
w3 = web3.Web3(web3.Web3.HTTPProvider(args.p))
|
||
|
||
f = open(os.path.join(args.abi_dir, 'AccountsIndex.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('owner address {}'.format(w3.eth.defaultAccount))
|
||
|
||
c = w3.eth.contract(abi=abi, address=args.r)
|
||
|
||
if args.w != None:
|
||
for w in args.w:
|
||
logg.info('adding {} to write list'.format(w))
|
||
print(c.functions.addWriter(w).transact().hex())
|
||
|
||
if args.a != None:
|
||
for a in args.a:
|
||
logg.info('adding {} to accounts index'.format(a))
|
||
print(c.functions.add(a).transact().hex())
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|