chaintool/ethd-gas-sum
4
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Allow separate settings for sender and recipient

Browse Source
This commit is contained in:
nolash 2021-10-28 13:35:51 +02:00
parent de201bc199
commit 674b5b9ae0
Signed by: lash
GPG Key ID: 21D2E7BB88C2A746
1 changed files with 37 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
eth_gas_sum/runnable/sum.py
View File

@ -33,13 +33,17 @@ argparser.add_argument('--start', type=int, help='start at block')
argparser.add_argument('--end', type=int, help='end block (not inclusive)') argparser.add_argument('--end', type=int, help='end block (not inclusive)')
argparser.add_argument('--interval', type=int, default=5, help='syncer poll interval for new blocks') argparser.add_argument('--interval', type=int, default=5, help='syncer poll interval for new blocks')
argparser.add_argument('-d', type=str, required=True, help='output directory') argparser.add_argument('-d', type=str, required=True, help='output directory')
argparser.add_positional('address', type=str, append=True, help='address sender to monitor') argparser.add_argument('--sender', type=str, action='append', default=[], help='sender address sender to monitor')
argparser.add_argument('--recipient', type=str, action='append', default=[], help='recipient address sender to monitor')
argparser.add_argument('--address', type=str, action='append', default=[], help='sender or recipient address to monitor')
args = argparser.parse_args() args = argparser.parse_args()
extra_args = { extra_args = {
'start': None, 'start': None,
'end': None, 'end': None,
'address': None, 'address': None,
'sender': None,
'recipient': None,
'd': '_OUTPUT_DIR', 'd': '_OUTPUT_DIR',
'interval': 'SYNCER_LOOP_INTERVAL', 'interval': 'SYNCER_LOOP_INTERVAL',
} }
@ -65,22 +69,25 @@ class EthChainInterface(ChainInterface):
class GasAddFilter: class GasAddFilter:
def __init__(self, chain_spec, addresses): def __init__(self, chain_spec, senders, recipients):
self.addresses = [] self.senders = senders
for address in addresses: self.recipients = recipients
clean_address = hex_uniform(strip_0x(address))
self.addresses.append(clean_address)
logg.debug('added {} to gas sum filter'.format(clean_address))
self.tx_gas = {} self.tx_gas = {}
self.gas_sum = 0 self.gas_sum = 0
def filter(self, conn, block, tx, db_session): def filter(self, conn, block, tx, db_session):
sender = hex_uniform(strip_0x(tx.outputs[0])) sender = hex_uniform(strip_0x(tx.outputs[0]))
if sender in self.addresses: recipient = hex_uniform(strip_0x(tx.inputs[0]))
if sender in self.senders:
self.gas_sum += tx.gas_used self.gas_sum += tx.gas_used
self.tx_gas[tx.hash] = tx.gas_used self.tx_gas[tx.hash] = tx.gas_used
logg.info('sender {} tx {} gas {} new sum {}'.format(sender, tx.hash, tx.gas_used, self.gas_sum)) logg.info('sender {} tx {} ({}/{}) gas {} new sum {}'.format(sender, tx.hash, tx.block.number, tx.index, tx.gas_used, self.gas_sum))
elif recipient in self.recipients:
self.gas_sum += tx.gas_used
self.tx_gas[tx.hash] = tx.gas_used
logg.info('recipient {} tx {} ({}/{}) gas {} new sum {}'.format(recipient, tx.hash, tx.block.number, tx.index, tx.gas_used, self.gas_sum))
def sum(self): def sum(self):
@ -106,7 +113,27 @@ def main():
backend = FileBackend.live(chain_spec, start, base_dir=config.get('_OUTPUT_DIR')) backend = FileBackend.live(chain_spec, start, base_dir=config.get('_OUTPUT_DIR'))
syncer = HeadSyncer(backend, chain_interface) syncer = HeadSyncer(backend, chain_interface)
gas_filter = GasAddFilter(chain_spec, config.get('_ADDRESS')) senders = []
recipients = []
for address in config.get('_SENDER'):
clean_address = hex_uniform(strip_0x(address))
senders.append(clean_address)
logg.debug('monitoring sender {}'.format(clean_address))
for address in config.get('_RECIPIENT'):
clean_address = hex_uniform(strip_0x(address))
recipients.append(clean_address)
logg.debug('monitoring recipient {}'.format(clean_address))
for address in config.get('_ADDRESS'):
clean_address = hex_uniform(strip_0x(address))
if address not in senders:
senders.append(clean_address)
logg.debug('monitoring sender {}'.format(clean_address))
if address not in recipients:
recipients.append(clean_address)
logg.debug('monitoring recipient {}'.format(clean_address))
gas_filter = GasAddFilter(chain_spec, senders, recipients)
syncer.add_filter(gas_filter) syncer.add_filter(gas_filter)
r = syncer.loop(config.get('SYNCER_LOOP_INTERVAL'), conn) r = syncer.loop(config.get('SYNCER_LOOP_INTERVAL'), conn)