class ExportTokensJob(BaseJob):
def __init__(self, web3, item_exporter, token_addresses_iterable,
max_workers):
self.item_exporter = item_exporter
self.token_addresses_iterable = token_addresses_iterable
self.batch_work_executor = BatchWorkExecutor(1, max_workers)
self.token_service = EthTokenService(web3, clean_user_provided_content)
self.token_mapper = EthTokenMapper()
def _start(self):
self.item_exporter.open()
def _export(self):
self.batch_work_executor.execute(self.token_addresses_iterable,
self._export_tokens)
def _export_tokens(self, token_addresses):
for token_address in token_addresses:
self._export_token(token_address)
def _export_token(self, token_address, block_number=None):
token = self.token_service.get_token(token_address)
token.block_number = block_number
token_dict = self.token_mapper.token_to_dict(token)
self.item_exporter.export_item(token_dict)
def _end(self):
self.batch_work_executor.shutdown()
self.item_exporter.close()
class ExtractGethTracesJob(BaseJob):
def __init__(self, traces_iterable, batch_size, max_workers,
item_exporter):
self.traces_iterable = traces_iterable
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.trace_mapper = EthTraceMapper()
self.geth_trace_mapper = EthGethTraceMapper()
def _start(self):
self.item_exporter.open()
def _export(self):
self.batch_work_executor.execute(self.traces_iterable,
self._extract_geth_traces)
def _extract_geth_traces(self, geth_traces):
for geth_trace_dict in geth_traces:
geth_trace = self.geth_trace_mapper.json_dict_to_geth_trace(
geth_trace_dict)
traces = self.trace_mapper.geth_trace_to_traces(geth_trace)
for trace in traces:
self.item_exporter.export_item(
self.trace_mapper.trace_to_dict(trace))
def _end(self):
self.batch_work_executor.shutdown()
self.item_exporter.close()
class ExportTokenTransfersJob(BaseJob):
def __init__(self,
start_block,
end_block,
batch_size,
web3,
item_exporter,
max_workers,
tokens=None):
validate_range(start_block, end_block)
self.start_block = start_block
self.end_block = end_block
self.web3 = web3
self.tokens = tokens
self.item_exporter = item_exporter
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.receipt_log_mapper = EthReceiptLogMapper()
self.token_transfer_mapper = EthTokenTransferMapper()
self.token_transfer_extractor = EthTokenTransferExtractor()
def _start(self):
self.item_exporter.open()
def _export(self):
self.batch_work_executor.execute(
range(self.start_block, self.end_block + 1),
self._export_batch,
total_items=self.end_block - self.start_block + 1)
def _export_batch(self, block_number_batch):
assert len(block_number_batch) > 0
# https://github.com/ethereum/wiki/wiki/JSON-RPC#eth_getfilterlogs
filter_params = {
'fromBlock': block_number_batch[0],
'toBlock': block_number_batch[-1],
'topics': [TRANSFER_EVENT_TOPIC]
}
if self.tokens is not None and len(self.tokens) > 0:
filter_params['address'] = self.tokens
event_filter = self.web3.eth.filter(filter_params)
events = event_filter.get_all_entries()
for event in events:
log = self.receipt_log_mapper.web3_dict_to_receipt_log(event)
token_transfer = self.token_transfer_extractor.extract_transfer_from_log(
log)
if token_transfer is not None:
self.item_exporter.export_item(
self.token_transfer_mapper.token_transfer_to_dict(
token_transfer))
self.web3.eth.uninstallFilter(event_filter.filter_id)
def _end(self):
self.batch_work_executor.shutdown()
self.item_exporter.close()
class ExtractTokenTransfersJob(BaseJob):
def __init__(self, logs_iterable, batch_size, max_workers, item_exporter):
self.logs_iterable = logs_iterable
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.receipt_log_mapper = EthReceiptLogMapper()
self.token_transfer_mapper = EthTokenTransferMapper()
self.token_transfer_extractor = EthTokenTransferExtractor()
def _start(self):
self.item_exporter.open()
def _export(self):
self.batch_work_executor.execute(self.logs_iterable,
self._extract_transfers)
def _extract_transfers(self, log_dicts):
for log_dict in log_dicts:
self._extract_transfer(log_dict)
def _extract_transfer(self, log_dict):
log = self.receipt_log_mapper.dict_to_receipt_log(log_dict)
token_transfer = self.token_transfer_extractor.extract_transfer_from_log(
log)
if token_transfer is not None:
self.item_exporter.export_item(
self.token_transfer_mapper.token_transfer_to_dict(
token_transfer))
def _end(self):
self.batch_work_executor.shutdown()
self.item_exporter.close()
def __init__(self, web3, item_exporter, token_addresses_iterable,
max_workers):
self.item_exporter = item_exporter
self.token_addresses_iterable = token_addresses_iterable
self.batch_work_executor = BatchWorkExecutor(1, max_workers)
self.token_service = EthTokenService(web3, clean_user_provided_content)
self.token_mapper = EthTokenMapper()
def __init__(self, logs_iterable, batch_size, max_workers, item_exporter):
self.logs_iterable = logs_iterable
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.receipt_log_mapper = EthReceiptLogMapper()
self.token_transfer_mapper = EthTokenTransferMapper()
self.token_transfer_extractor = EthTokenTransferExtractor()
def __init__(self, traces_iterable, batch_size, max_workers,
item_exporter):
self.traces_iterable = traces_iterable
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.trace_mapper = EthTraceMapper()
self.geth_trace_mapper = EthGethTraceMapper()
def __init__(self, traces_iterable, batch_size, max_workers,
item_exporter):
self.traces_iterable = traces_iterable
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.contract_service = EthContractService()
self.contract_mapper = EthContractMapper()
class ExportContractsJob(BaseJob):
def __init__(self, contract_addresses_iterable, batch_size,
batch_web3_provider, max_workers, item_exporter):
self.batch_web3_provider = batch_web3_provider
self.contract_addresses_iterable = contract_addresses_iterable
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.contract_service = EthContractService()
self.contract_mapper = EthContractMapper()
def _start(self):
self.item_exporter.open()
def _export(self):
self.batch_work_executor.execute(self.contract_addresses_iterable,
self._export_contracts)
def _export_contracts(self, contract_addresses):
contracts_code_rpc = list(
generate_get_code_json_rpc(contract_addresses))
response_batch = self.batch_web3_provider.make_batch_request(
json.dumps(contracts_code_rpc))
contracts = []
for response in response_batch:
# request id is the index of the contract address in contract_addresses list
request_id = response['id']
result = rpc_response_to_result(response)
contract_address = contract_addresses[request_id]
contract = self._get_contract(contract_address, result)
contracts.append(contract)
for contract in contracts:
self.item_exporter.export_item(
self.contract_mapper.contract_to_dict(contract))
def _get_contract(self, contract_address, rpc_result):
contract = self.contract_mapper.rpc_result_to_contract(
contract_address, rpc_result)
bytecode = contract.bytecode
function_sighashes = self.contract_service.get_function_sighashes(
bytecode)
contract.function_sighashes = function_sighashes
contract.is_erc20 = self.contract_service.is_erc20_contract(
function_sighashes)
contract.is_erc721 = self.contract_service.is_erc721_contract(
function_sighashes)
return contract
def _end(self):
self.batch_work_executor.shutdown()
self.item_exporter.close()
def __init__(self, contract_addresses_iterable, batch_size,
batch_web3_provider, max_workers, item_exporter):
self.batch_web3_provider = batch_web3_provider
self.contract_addresses_iterable = contract_addresses_iterable
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.contract_service = EthContractService()
self.contract_mapper = EthContractMapper()
class ExtractContractsJob(BaseJob):
def __init__(self, traces_iterable, batch_size, max_workers,
item_exporter):
self.traces_iterable = traces_iterable
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.contract_service = EthContractService()
self.contract_mapper = EthContractMapper()
def _start(self):
self.item_exporter.open()
def _export(self):
self.batch_work_executor.execute(self.traces_iterable,
self._extract_contracts)
def _extract_contracts(self, traces):
for trace in traces:
trace['status'] = to_int_or_none(trace.get('status'))
trace['block_number'] = to_int_or_none(trace.get('block_number'))
contract_creation_traces = [
trace for trace in traces if trace.get('trace_type') == 'create'
and trace.get('to_address') is not None
and len(trace.get('to_address')) > 0 and trace.get('status') == 1
]
contracts = []
for trace in contract_creation_traces:
contract = EthContract()
contract.address = trace.get('to_address')
bytecode = trace.get('output')
contract.bytecode = bytecode
contract.block_number = trace.get('block_number')
function_sighashes = self.contract_service.get_function_sighashes(
bytecode)
contract.function_sighashes = function_sighashes
contract.is_erc20 = self.contract_service.is_erc20_contract(
function_sighashes)
contract.is_erc721 = self.contract_service.is_erc721_contract(
function_sighashes)
contracts.append(contract)
for contract in contracts:
self.item_exporter.export_item(
self.contract_mapper.contract_to_dict(contract))
def _end(self):
self.batch_work_executor.shutdown()
self.item_exporter.close()
class ExportReceiptsJob(BaseJob):
def __init__(self,
transaction_hashes_iterable,
batch_size,
batch_web3_provider,
max_workers,
item_exporter,
export_receipts=True,
export_logs=True):
self.batch_web3_provider = batch_web3_provider
self.transaction_hashes_iterable = transaction_hashes_iterable
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.export_receipts = export_receipts
self.export_logs = export_logs
if not self.export_receipts and not self.export_logs:
raise ValueError(
'At least one of export_receipts or export_logs must be True')
self.receipt_mapper = EthReceiptMapper()
self.receipt_log_mapper = EthReceiptLogMapper()
def _start(self):
self.item_exporter.open()
def _export(self):
self.batch_work_executor.execute(self.transaction_hashes_iterable,
self._export_receipts)
def _export_receipts(self, transaction_hashes):
receipts_rpc = list(generate_get_receipt_json_rpc(transaction_hashes))
response = self.batch_web3_provider.make_batch_request(
json.dumps(receipts_rpc))
results = rpc_response_batch_to_results(response)
receipts = [
self.receipt_mapper.json_dict_to_receipt(result)
for result in results
]
for receipt in receipts:
self._export_receipt(receipt)
def _export_receipt(self, receipt):
if self.export_receipts:
self.item_exporter.export_item(
self.receipt_mapper.receipt_to_dict(receipt))
if self.export_logs:
for log in receipt.logs:
self.item_exporter.export_item(
self.receipt_log_mapper.receipt_log_to_dict(log))
def _end(self):
self.batch_work_executor.shutdown()
self.item_exporter.close()
def __init__(self, start_block, end_block, batch_size, batch_web3_provider,
max_workers, item_exporter):
validate_range(start_block, end_block)
self.start_block = start_block
self.end_block = end_block
self.batch_web3_provider = batch_web3_provider
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.geth_trace_mapper = EthGethTraceMapper()
class ExportGethTracesJob(BaseJob):
def __init__(self, start_block, end_block, batch_size, batch_web3_provider,
max_workers, item_exporter):
validate_range(start_block, end_block)
self.start_block = start_block
self.end_block = end_block
self.batch_web3_provider = batch_web3_provider
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.geth_trace_mapper = EthGethTraceMapper()
def _start(self):
self.item_exporter.open()
def _export(self):
self.batch_work_executor.execute(
range(self.start_block, self.end_block + 1),
self._export_batch,
total_items=self.end_block - self.start_block + 1)
def _export_batch(self, block_number_batch):
trace_block_rpc = list(
generate_trace_block_by_number_json_rpc(block_number_batch))
response = self.batch_web3_provider.make_batch_request(
json.dumps(trace_block_rpc))
for response_item in response:
block_number = response_item.get('id')
result = rpc_response_to_result(response_item)
geth_trace = self.geth_trace_mapper.json_dict_to_geth_trace({
'block_number':
block_number,
'transaction_traces':
[tx_trace.get('result') for tx_trace in result],
})
self.item_exporter.export_item(
self.geth_trace_mapper.geth_trace_to_dict(geth_trace))
def _end(self):
self.batch_work_executor.shutdown()
self.item_exporter.close()
def __init__(self,
start_block,
end_block,
batch_size,
web3,
item_exporter,
max_workers,
tokens=None):
validate_range(start_block, end_block)
self.start_block = start_block
self.end_block = end_block
self.web3 = web3
self.tokens = tokens
self.item_exporter = item_exporter
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.receipt_log_mapper = EthReceiptLogMapper()
self.token_transfer_mapper = EthTokenTransferMapper()
self.token_transfer_extractor = EthTokenTransferExtractor()
def __init__(self,
transaction_hashes_iterable,
batch_size,
batch_web3_provider,
max_workers,
item_exporter,
export_receipts=True,
export_logs=True):
self.batch_web3_provider = batch_web3_provider
self.transaction_hashes_iterable = transaction_hashes_iterable
self.batch_work_executor = BatchWorkExecutor(batch_size, max_workers)
self.item_exporter = item_exporter
self.export_receipts = export_receipts
self.export_logs = export_logs
if not self.export_receipts and not self.export_logs:
raise ValueError(
'At least one of export_receipts or export_logs must be True')
self.receipt_mapper = EthReceiptMapper()
self.receipt_log_mapper = EthReceiptLogMapper()
def __init__(self,
start_block,
end_block,
batch_size,
web3,
item_exporter,
max_workers,
include_genesis_traces=False,
include_daofork_traces=False):
validate_range(start_block, end_block)
self.start_block = start_block
self.end_block = end_block
self.web3 = web3
# TODO: use batch_size when this issue is fixed https://github.com/paritytech/parity-ethereum/issues/9822
self.batch_work_executor = BatchWorkExecutor(1, max_workers)
self.item_exporter = item_exporter
self.trace_mapper = EthTraceMapper()
self.special_trace_service = EthSpecialTraceService()
self.include_genesis_traces = include_genesis_traces
self.include_daofork_traces = include_daofork_traces
class ExportTracesJob(BaseJob):
def __init__(self,
start_block,
end_block,
batch_size,
web3,
item_exporter,
max_workers,
include_genesis_traces=False,
include_daofork_traces=False):
validate_range(start_block, end_block)
self.start_block = start_block
self.end_block = end_block
self.web3 = web3
# TODO: use batch_size when this issue is fixed https://github.com/paritytech/parity-ethereum/issues/9822
self.batch_work_executor = BatchWorkExecutor(1, max_workers)
self.item_exporter = item_exporter
self.trace_mapper = EthTraceMapper()
self.special_trace_service = EthSpecialTraceService()
self.include_genesis_traces = include_genesis_traces
self.include_daofork_traces = include_daofork_traces
def _start(self):
self.item_exporter.open()
def _export(self):
self.batch_work_executor.execute(
range(self.start_block, self.end_block + 1),
self._export_batch,
total_items=self.end_block - self.start_block + 1)
def _export_batch(self, block_number_batch):
# TODO: Change to len(block_number_batch) > 0 when this issue is fixed
# https://github.com/paritytech/parity-ethereum/issues/9822
assert len(block_number_batch) == 1
block_number = block_number_batch[0]
all_traces = []
if self.include_genesis_traces and 0 in block_number_batch:
genesis_traces = self.special_trace_service.get_genesis_traces()
all_traces.extend(genesis_traces)
if self.include_daofork_traces and DAOFORK_BLOCK_NUMBER in block_number_batch:
daofork_traces = self.special_trace_service.get_daofork_traces()
all_traces.extend(daofork_traces)
# TODO: Change to traceFilter when this issue is fixed
# https://github.com/paritytech/parity-ethereum/issues/9822
json_traces = self.web3.parity.traceBlock(block_number)
if json_traces is None:
raise ValueError(
'Response from the node is None. Is the node fully synced?')
traces = [
self.trace_mapper.json_dict_to_trace(json_trace)
for json_trace in json_traces
]
all_traces.extend(traces)
calculate_trace_statuses(all_traces)
calculate_trace_ids(all_traces)
for trace in all_traces:
self.item_exporter.export_item(
self.trace_mapper.trace_to_dict(trace))
def _end(self):
self.batch_work_executor.shutdown()
self.item_exporter.close()