65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
# external imports
|
|
from hexathon import add_0x
|
|
|
|
# local imports
|
|
from chainqueue.enum import (
|
|
StatusBits,
|
|
all_errors,
|
|
is_alive,
|
|
is_error_status,
|
|
status_str,
|
|
)
|
|
|
|
|
|
class Outputter:
|
|
|
|
def __init__(self, chain_spec, writer, getter, session_method=None, decode_status=True):
|
|
self.decode_status = decode_status
|
|
self.writer = writer
|
|
self.getter = getter
|
|
self.chain_spec = chain_spec
|
|
self.chain_spec_str = str(chain_spec)
|
|
self.session = None
|
|
if session_method != None:
|
|
self.session = session_method()
|
|
self.results = {
|
|
'pending_error': 0,
|
|
'final_error': 0,
|
|
'pending': 0,
|
|
'final': 0,
|
|
}
|
|
|
|
|
|
def __del__(self):
|
|
if self.session != None:
|
|
self.session.close()
|
|
|
|
|
|
def add(self, tx_hash):
|
|
tx = tx_getter(self.chain_spec, tx_hash, session=self.session)
|
|
category = None
|
|
if is_alive(tx['status_code']):
|
|
category = 'pending'
|
|
else:
|
|
category = 'final'
|
|
self.results[category] += 1
|
|
if is_error_status(tx['status_code']):
|
|
logg.debug('registered {} as {} with error'.format(tx_hash, category))
|
|
self.results[category + '_error'] += 1
|
|
else:
|
|
logg.debug('registered {} as {}'.format(tx_hash, category))
|
|
|
|
|
|
def decode_summary(self):
|
|
self.writer.write('pending\t{}\t{}\n'.format(self.results['pending'], self.results['pending_error']))
|
|
self.writer.write('final\t{}\t{}\n'.format(self.results['final'], self.results['final_error']))
|
|
self.writer.write('total\t{}\t{}\n'.format(self.results['final'] + self.results['pending'], self.results['final_error'] + self.results['pending_error']))
|
|
|
|
|
|
def decode_single(self, tx_hash):
|
|
tx = self.getter(self.chain_spec, tx_hash, session=self.session)
|
|
status = tx['status']
|
|
if self.decode_status:
|
|
status = status_str(tx['status_code'], bits_only=True)
|
|
self.writer.write('{}\t{}\t{}\t{}\n'.format(self.chain_spec_str, add_0x(tx_hash), status, tx['status_code']))
|