91 lines
2.9 KiB
Python
91 lines
2.9 KiB
Python
|
# standard imports
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
# external imports
|
|||
|
|
from chainlib.connection import (
|
|||
|
|
RPCConnection,
|
|||
|
|
ConnType,
|
|||
|
|
)
|
|||
|
|
from chainlib.eth.connection import (
|
|||
|
|
EthUnixSignerConnection,
|
|||
|
|
EthHTTPSignerConnection,
|
|||
|
|
)
|
|||
|
|
from chainlib.chain import ChainSpec
|
|||
|
|
|
|||
|
|
logg = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class RPC:
|
|||
|
|
|
|||
|
|
def __init__(self, chain_spec, rpc_provider, signer_provider=None):
|
|||
|
|
self.chain_spec = chain_spec
|
|||
|
|
self.rpc_provider = rpc_provider
|
|||
|
|
self.signer_provider = signer_provider
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_default(self):
|
|||
|
|
return self.get_by_label('default')
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_by_label(self, label):
|
|||
|
|
return RPCConnection.connect(self.chain_spec, label)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def from_config(config, use_signer=False, default_label='default', signer_label='signer'):
|
|||
|
|
chain_spec = ChainSpec.from_chain_str(config.get('CHAIN_SPEC'))
|
|||
|
|
|
|||
|
|
RPCConnection.register_location(config.get('RPC_PROVIDER'), chain_spec, default_label)
|
|||
|
|
if use_signer:
|
|||
|
|
|
|||
|
|
RPCConnection.register_constructor(ConnType.UNIX, EthUnixSignerConnection, signer_label)
|
|||
|
|
RPCConnection.register_constructor(ConnType.HTTP, EthHTTPSignerConnection, signer_label)
|
|||
|
|
RPCConnection.register_constructor(ConnType.HTTP_SSL, EthHTTPSignerConnection, signer_label)
|
|||
|
|
RPCConnection.register_location(config.get('SIGNER_PROVIDER'), chain_spec, signer_label)
|
|||
|
|
rpc = RPC(chain_spec, config.get('RPC_PROVIDER'), signer_provider=config.get('SIGNER_PROVIDER'))
|
|||
|
|
logg.info('set up rpc: {}'.format(rpc))
|
|||
|
|
return rpc
|
|||
|
|
|
|||
|
|
|
|||
|
|
def __str__(self):
|
|||
|
|
return 'RPC factory, chain {}, rpc {}, signer {}'.format(self.chain_spec, self.rpc_provider, self.signer_provider)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# TOOD: re-implement file backend option from omittec code:
|
|||
|
|
#broker = config.get('CELERY_BROKER_URL')
|
|||
|
|
#if broker[:4] == 'file':
|
|||
|
|
# bq = tempfile.mkdtemp()
|
|||
|
|
# bp = tempfile.mkdtemp()
|
|||
|
|
# conf_update = {
|
|||
|
|
# 'broker_url': broker,
|
|||
|
|
# 'broker_transport_options': {
|
|||
|
|
# 'data_folder_in': bq,
|
|||
|
|
# 'data_folder_out': bq,
|
|||
|
|
# 'data_folder_processed': bp,
|
|||
|
|
# },
|
|||
|
|
# }
|
|||
|
|
# if config.true('CELERY_DEBUG'):
|
|||
|
|
# conf_update['result_extended'] = True
|
|||
|
|
# current_app.conf.update(conf_update)
|
|||
|
|
# logg.warning('celery broker dirs queue i/o {} processed {}, will NOT be deleted on shutdown'.format(bq, bp))
|
|||
|
|
#else:
|
|||
|
|
# conf_update = {
|
|||
|
|
# 'broker_url': broker,
|
|||
|
|
# }
|
|||
|
|
# if config.true('CELERY_DEBUG'):
|
|||
|
|
# conf_update['result_extended'] = True
|
|||
|
|
# current_app.conf.update(conf_update)
|
|||
|
|
#
|
|||
|
|
#result = config.get('CELERY_RESULT_URL')
|
|||
|
|
#if result[:4] == 'file':
|
|||
|
|
# rq = tempfile.mkdtemp()
|
|||
|
|
# current_app.conf.update({
|
|||
|
|
# 'result_backend': 'file://{}'.format(rq),
|
|||
|
|
# })
|
|||
|
|
# logg.warning('celery backend store dir {} created, will NOT be deleted on shutdown'.format(rq))
|
|||
|
|
#else:
|
|||
|
|
# current_app.conf.update({
|
|||
|
|
# 'result_backend': result,
|
|||
|
|
# })
|
|||
|
|
#
|