def test_export_erc20_tokens_job(tmpdir, token_addresses, resource_group):
output_file = tmpdir.join('erc20_tokens.csv')
job = ExportErc20TokensJob(
token_addresses_iterable=token_addresses,
web3=ThreadLocalProxy(lambda: Web3(MockWeb3Provider(resource_group))),
item_exporter=erc20_tokens_item_exporter(output_file),
max_workers=5)
job.run()
compare_lines_ignore_order(
read_resource(resource_group, 'expected_erc20_tokens.csv'),
read_file(output_file))
parser.add_argument('-o',
'--output',
default='-',
type=str,
help='The output file. If not specified stdout is used.')
parser.add_argument('-w',
'--max-workers',
default=5,
type=int,
help='The maximum number of workers.')
parser.add_argument(
'-p',
'--provider-uri',
default=None,
type=str,
help='The URI of the web3 provider e.g. '
'file://$HOME/Library/Ethereum/geth.ipc or https://mainnet.infura.io/')
args = parser.parse_args()
with smart_open(args.token_addresses, 'r') as token_addresses_file:
job = ExportErc20TokensJob(
token_addresses_iterable=(token_address.strip()
for token_address in token_addresses_file),
web3=ThreadLocalProxy(
lambda: Web3(get_provider_from_uri(args.provider_uri))),
item_exporter=export_erc20_tokens_job_item_exporter(args.output),
max_workers=args.max_workers)
job.run()
#!/usr/bin/env python3
import argparse
from web3 import IPCProvider, Web3
from ethereumetl.file_utils import smart_open
from ethereumetl.jobs.export_erc20_tokens_job import ExportErc20TokensJob
from ethereumetl.jobs.export_erc20_tokens_job_item_exporter import export_erc20_tokens_job_item_exporter
from ethereumetl.thread_local_proxy import ThreadLocalProxy
parser = argparse.ArgumentParser(description='Exports ERC20 tokens.')
parser.add_argument('-t', '--token-addresses', type=str, help='The file containing token addresses, one per line.')
parser.add_argument('-o', '--output', default='-', type=str, help='The output file. If not specified stdout is used.')
parser.add_argument('-w', '--max-workers', default=5, type=int, help='The maximum number of workers.')
parser.add_argument('--ipc-path', required=True, type=str, help='The full path to the ipc socket file.')
parser.add_argument('--ipc-timeout', default=60, type=int, help='The timeout in seconds for ipc calls.')
args = parser.parse_args()
with smart_open(args.token_addresses, 'r') as token_addresses_file:
job = ExportErc20TokensJob(
token_addresses_iterable=(token_address.strip() for token_address in token_addresses_file),
web3=ThreadLocalProxy(lambda: Web3(IPCProvider(args.ipc_path, timeout=args.ipc_timeout))),
item_exporter=export_erc20_tokens_job_item_exporter(args.output),
max_workers=args.max_workers)
job.run()