mirror of
git://holbrook.no/eth-address-index
synced 2024-11-19 15:46:46 +01:00
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
"""Adds a new token to the token symbol index
|
||
|
||
.. moduleauthor:: Louis Holbrook <dev@holbrook.no>
|
||
.. pgp:: 0826EDA1702D1E87C6E2875121D2E7BB88C2A746
|
||
|
||
"""
|
||
|
||
# standard imports
|
||
import os
|
||
import json
|
||
import argparse
|
||
import logging
|
||
import hashlib
|
||
|
||
# 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='Token endorsement contract address')
|
||
argparser.add_argument('-o', '--owner-address', dest='o', type=str, help='Address to use to sign endorsement transaction')
|
||
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')
|
||
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, 'TokenUniqueSymbolIndex.json'), 'r')
|
||
abi = json.load(f)
|
||
f.close()
|
||
|
||
f = open(os.path.join(args.abi_dir, 'ERC20.json'), 'r')
|
||
erc20_abi = json.load(f)
|
||
f.close()
|
||
|
||
t = w3.eth.contract(abi=erc20_abi, address=args.address)
|
||
token_symbol = t.functions.symbol().call()
|
||
|
||
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)
|
||
|
||
h = hashlib.new('sha256')
|
||
h.update(token_symbol.encode('utf-8'))
|
||
z = h.digest()
|
||
logg.info('token symbol {} => {}'.format(token_symbol, z.hex()))
|
||
|
||
tx_hash = c.functions.register('0x' + z.hex(), args.address).transact()
|
||
print(tx_hash.hex())
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main()
|