mirror of git://holbrook.no/eth-monitor.git
144 lines
5.1 KiB
Python
144 lines
5.1 KiB
Python
# standard imports
|
||
import logging
|
||
import os
|
||
import uuid
|
||
import importlib
|
||
|
||
# external imports
|
||
from chainlib.settings import ChainSettings
|
||
from chainsyncer.settings import ChainsyncerSettings
|
||
from chainlib.eth.connection import EthHTTPConnection
|
||
|
||
# local imports
|
||
from eth_monitor.rules import (
|
||
AddressRules,
|
||
RuleSimple,
|
||
RuleMethod,
|
||
RuleData,
|
||
)
|
||
|
||
logg = logging.getLogger(__name__)
|
||
|
||
|
||
class EthMonitorSettings(ChainsyncerSettings):
|
||
|
||
def process_monitor_session(self, config):
|
||
session_id = config.get('_SESSION_ID')
|
||
if session_id == None:
|
||
if config.get('_SINGLE'):
|
||
session_id = str(uuid.uuid4())
|
||
else:
|
||
session_id = 'default'
|
||
|
||
self.o['SESSION_ID'] = session_id
|
||
|
||
|
||
def process_monitor_session_dir(self, config):
|
||
syncer_store_module = None
|
||
syncer_store_class = None
|
||
state_dir = None
|
||
if config.get('SYNCER_BACKEND') == 'mem':
|
||
syncer_store_module = importlib.import_module('chainsyncer.store.mem')
|
||
syncer_store_class = getattr(syncer_store_module, 'SyncMemStore')
|
||
else:
|
||
if config.get('SYNCER_BACKEND') == 'fs':
|
||
syncer_store_module = importlib.import_module('chainsyncer.store.fs')
|
||
syncer_store_class = getattr(syncer_store_module, 'SyncFsStore')
|
||
elif config.get('SYNCER_BACKEND') == 'rocksdb':
|
||
syncer_store_module = importlib.import_module('chainsyncer.store.rocksdb')
|
||
syncer_store_class = getattr(syncer_store_module, 'SyncRocksDbStore')
|
||
else:
|
||
syncer_store_module = importlib.import_module(config.get('SYNCER_BACKEND'))
|
||
syncer_store_class = getattr(syncer_store_module, 'SyncStore')
|
||
state_dir = os.path.join(config.get('_STATE_DIR'), config.get('SYNCER_BACKEND'))
|
||
|
||
logg.info('using engine {} module {}.{}'.format(config.get('SYNCER_BACKEND'), syncer_store_module.__file__, syncer_store_class.__name__))
|
||
|
||
#state_dir = os.path.join(state_dir, config.get('_SESSION_ID'))
|
||
sync_store = syncer_store_class(state_dir, session_id=session.get('SESSION_ID'), state_event_callback=state_change_callback, filter_state_event_callback=filter_change_callback)
|
||
|
||
self.o['STATE_DIR'] = state_dir
|
||
self.o['SYNC_STORE'] = sync_store
|
||
|
||
|
||
#def process_address_arg_rules(rules, args):
|
||
def process_address_arg_rules(self, config):
|
||
include_inputs = config.get('ETHMONITOR_INPUTS')
|
||
if include_inputs == None:
|
||
include_inputs = []
|
||
else:
|
||
include_inputs = include_inputs.split(',')
|
||
|
||
include_outputs = config.get('ETHMONITOR_OUTPUTS')
|
||
if include_outputs == None:
|
||
include_outputs = []
|
||
else:
|
||
include_outputs = include_outputs.split(',')
|
||
|
||
include_exec = config.get('ETHMONITOR_EXEC')
|
||
if include_exec == None:
|
||
include_exec = []
|
||
else:
|
||
include_exec = include_exec.split(',')
|
||
|
||
exclude_inputs = config.get('ETHMONITOR_X_INPUTS')
|
||
if exclude_inputs == None:
|
||
exclude_inputs = []
|
||
else:
|
||
exclude_inputs = exclude_inputs.split(',')
|
||
|
||
exclude_outputs = config.get('ETHMONITOR_X_OUTPUTS')
|
||
if exclude_outputs == None:
|
||
exclude_outputs = []
|
||
else:
|
||
exclude_outputs = exclude_outputs.split(',')
|
||
|
||
exclude_exec = config.get('ETHMONITOR_X_EXEC')
|
||
if exclude_exec == None:
|
||
exclude_exec = []
|
||
else:
|
||
exclude_exec = exclude_exec.split(',')
|
||
|
||
|
||
address = config.get('ETHMONITOR_ADDRESS')
|
||
if address != None:
|
||
for address in address.split(','):
|
||
include_inputs.append(address)
|
||
include_outputs.append(address)
|
||
include_exec.append(address)
|
||
|
||
address = config.get('ETHMONITOR_X_ADDRESS')
|
||
if address != None:
|
||
for address in address.split(','):
|
||
exclude_inputs.append(address)
|
||
exclude_outputs.append(address)
|
||
exclude_exec.append(address)
|
||
|
||
includes = RuleSimple(include_outputs, include_inputs, include_exec, description='INCLUDE')
|
||
self.o['RULES'].include(includes)
|
||
|
||
excludes = RuleSimple(exclude_outputs, exclude_inputs, exclude_exec, description='EXCLUDE')
|
||
self.o['RULES'].exclude(excludes)
|
||
|
||
|
||
def process_arg_rules(self, config):
|
||
address_rules = AddressRules(include_by_default=config.get('ETHMONITOR_INCLUDE_DEFAULT'))
|
||
self.o['RULES'] = address_rules
|
||
|
||
self.process_address_arg_rules(config)
|
||
|
||
|
||
def process_common(self, config):
|
||
super(EthMonitorSettings, self).process_common(config)
|
||
# TODO: duplicate from chaind, consider move to chainlib-eth
|
||
rpc_provider = config.get('RPC_PROVIDER')
|
||
if rpc_provider == None:
|
||
rpc_provider = 'http://localhost:8545'
|
||
self.o['RPC'] = EthHTTPConnection(url=rpc_provider, chain_spec=self.o['CHAIN_SPEC'])
|
||
|
||
|
||
def process(self, config):
|
||
self.process_common(config)
|
||
self.process_monitor_session(config)
|
||
self.process_arg_rules(config)
|