137 lines
3.9 KiB
Python
137 lines
3.9 KiB
Python
|
#!python3
|
|||
|
|
|||
|
"""Token balance query script
|
|||
|
|
|||
|
.. moduleauthor:: Louis Holbrook <dev@holbrook.no>
|
|||
|
.. pgp:: 0826EDA1702D1E87C6E2875121D2E7BB88C2A746
|
|||
|
|
|||
|
"""
|
|||
|
|
|||
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|||
|
|
|||
|
# standard imports
|
|||
|
import sys
|
|||
|
import os
|
|||
|
import json
|
|||
|
import argparse
|
|||
|
import logging
|
|||
|
|
|||
|
# third-party imports
|
|||
|
from hexathon import (
|
|||
|
add_0x,
|
|||
|
strip_0x,
|
|||
|
even,
|
|||
|
)
|
|||
|
import sha3
|
|||
|
from eth_abi import encode_single
|
|||
|
|
|||
|
# local imports
|
|||
|
from chainlib.eth.address import (
|
|||
|
to_checksum_address,
|
|||
|
is_checksum_address,
|
|||
|
)
|
|||
|
from chainlib.jsonrpc import (
|
|||
|
jsonrpc_template,
|
|||
|
jsonrpc_result,
|
|||
|
)
|
|||
|
from chainlib.eth.block import block_latest
|
|||
|
from chainlib.eth.tx import count
|
|||
|
from chainlib.eth.erc20 import ERC20
|
|||
|
from chainlib.eth.connection import EthHTTPConnection
|
|||
|
from chainlib.eth.gas import (
|
|||
|
OverrideGasOracle,
|
|||
|
balance,
|
|||
|
price,
|
|||
|
)
|
|||
|
from chainlib.chain import ChainSpec
|
|||
|
|
|||
|
logging.basicConfig(level=logging.WARNING)
|
|||
|
logg = logging.getLogger()
|
|||
|
|
|||
|
default_abi_dir = os.environ.get('ETH_ABI_DIR', '/usr/share/local/cic/solidity/abi')
|
|||
|
default_eth_provider = os.environ.get('ETH_PROVIDER', 'http://localhost:8545')
|
|||
|
|
|||
|
argparser = argparse.ArgumentParser()
|
|||
|
argparser.add_argument('-p', '--provider', dest='p', default=default_eth_provider, type=str, help='Web3 provider url (http only)')
|
|||
|
argparser.add_argument('-i', '--chain-spec', dest='i', type=str, default='evm:ethereum:1', help='Chain specification string')
|
|||
|
argparser.add_argument('-H', '--human', dest='human', action='store_true', help='Use human-friendly formatting')
|
|||
|
argparser.add_argument('-u', '--unsafe', dest='u', action='store_true', help='Auto-convert address to checksum adddress')
|
|||
|
argparser.add_argument('-v', action='store_true', help='Be verbose')
|
|||
|
argparser.add_argument('-vv', action='store_true', help='Be more verbose')
|
|||
|
argparser.add_argument('-y', '--key-file', dest='y', type=str, help='Include summary for keyfile')
|
|||
|
argparser.add_argument('address', nargs='?', type=str, help='Include summary for address (conflicts with -y)')
|
|||
|
args = argparser.parse_args()
|
|||
|
|
|||
|
|
|||
|
if args.vv:
|
|||
|
logg.setLevel(logging.DEBUG)
|
|||
|
elif args.v:
|
|||
|
logg.setLevel(logging.INFO)
|
|||
|
|
|||
|
signer = None
|
|||
|
holder_address = None
|
|||
|
if args.address != None:
|
|||
|
if not args.u and is_checksum_address(args.address):
|
|||
|
raise ValueError('invalid checksum address {}'.format(args.address))
|
|||
|
holder_address = add_0x(args.address)
|
|||
|
elif args.y != None:
|
|||
|
f = open(args.y, 'r')
|
|||
|
o = json.load(f)
|
|||
|
f.close()
|
|||
|
holder_address = add_0x(to_checksum_address(o['address']))
|
|||
|
|
|||
|
|
|||
|
#if holder_address != None:
|
|||
|
# passphrase_env = 'ETH_PASSPHRASE'
|
|||
|
# if args.env_prefix != None:
|
|||
|
# passphrase_env = args.env_prefix + '_' + passphrase_env
|
|||
|
# passphrase = os.environ.get(passphrase_env)
|
|||
|
# logg.error('pass {}'.format(passphrase_env))
|
|||
|
# if passphrase == None:
|
|||
|
# logg.warning('no passphrase given')
|
|||
|
# passphrase=''
|
|||
|
#
|
|||
|
# holder_address = None
|
|||
|
# keystore = DictKeystore()
|
|||
|
# if args.y != None:
|
|||
|
# logg.debug('loading keystore file {}'.format(args.y))
|
|||
|
# signer_address = keystore.import_keystore_file(args.y, password=passphrase)
|
|||
|
# logg.debug('now have key for signer address {}'.format(signer_address))
|
|||
|
# signer = EIP155Signer(keystore)
|
|||
|
|
|||
|
conn = EthHTTPConnection(args.p)
|
|||
|
gas_oracle = OverrideGasOracle(conn)
|
|||
|
|
|||
|
token_symbol = 'eth'
|
|||
|
|
|||
|
chain_spec = ChainSpec.from_chain_str(args.i)
|
|||
|
|
|||
|
human = args.human
|
|||
|
|
|||
|
|
|||
|
def main():
|
|||
|
o = block_latest()
|
|||
|
r = conn.do(o)
|
|||
|
n = int(r, 16)
|
|||
|
if human:
|
|||
|
n = format(n, ',')
|
|||
|
sys.stdout.write('Block: {}\n'.format(n))
|
|||
|
|
|||
|
o = price()
|
|||
|
r = conn.do(o)
|
|||
|
n = int(r, 16)
|
|||
|
if human:
|
|||
|
n = format(n, ',')
|
|||
|
sys.stdout.write('Gasprice: {}\n'.format(n))
|
|||
|
|
|||
|
if holder_address != None:
|
|||
|
o = count(holder_address)
|
|||
|
r = conn.do(o)
|
|||
|
n = int(r, 16)
|
|||
|
sys.stdout.write('Address: {}\n'.format(holder_address))
|
|||
|
sys.stdout.write('Nonce: {}\n'.format(n))
|
|||
|
|
|||
|
|
|||
|
if __name__ == '__main__':
|
|||
|
main()
|