55 lines
3.9 KiB
Python
55 lines
3.9 KiB
Python
|
from scribe.env import Env
|
||
|
|
|
||
|
|
|
||
|
|
class ElasticEnv(Env):
|
||
|
|
def __init__(self, db_dir=None, max_query_workers=None, chain=None, reorg_limit=None, prometheus_port=None,
|
||
|
|
cache_all_tx_hashes=None, cache_all_claim_txos=None, elastic_host=None, elastic_port=None,
|
||
|
|
es_index_prefix=None, elastic_notifier_host=None, elastic_notifier_port=None,
|
||
|
|
blocking_channel_ids=None, filtering_channel_ids=None, reindex=False):
|
||
|
|
super().__init__(db_dir, max_query_workers, chain, reorg_limit, prometheus_port, cache_all_tx_hashes,
|
||
|
|
cache_all_claim_txos, blocking_channel_ids, filtering_channel_ids)
|
||
|
|
self.elastic_host = elastic_host if elastic_host is not None else self.default('ELASTIC_HOST', 'localhost')
|
||
|
|
self.elastic_port = elastic_port if elastic_port is not None else self.integer('ELASTIC_PORT', 9200)
|
||
|
|
self.elastic_notifier_host = elastic_notifier_host if elastic_notifier_host is not None else self.default(
|
||
|
|
'ELASTIC_NOTIFIER_HOST', 'localhost')
|
||
|
|
self.elastic_notifier_port = elastic_notifier_port if elastic_notifier_port is not None else self.integer(
|
||
|
|
'ELASTIC_NOTIFIER_PORT', 19080)
|
||
|
|
self.es_index_prefix = es_index_prefix if es_index_prefix is not None else self.default('ES_INDEX_PREFIX', '')
|
||
|
|
# Filtering / Blocking
|
||
|
|
self.blocking_channel_ids = blocking_channel_ids if blocking_channel_ids is not None else self.default(
|
||
|
|
'BLOCKING_CHANNEL_IDS', '').split(' ')
|
||
|
|
self.filtering_channel_ids = filtering_channel_ids if filtering_channel_ids is not None else self.default(
|
||
|
|
'FILTERING_CHANNEL_IDS', '').split(' ')
|
||
|
|
self.reindex = reindex if reindex is not None else self.boolean('REINDEX_ES', False)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def contribute_to_arg_parser(cls, parser):
|
||
|
|
super().contribute_to_arg_parser(parser)
|
||
|
|
parser.add_argument('--reindex', default=False, help="Drop and rebuild the elasticsearch index.",
|
||
|
|
action='store_true')
|
||
|
|
parser.add_argument('--elastic_host', default=cls.default('ELASTIC_HOST', 'localhost'), type=str,
|
||
|
|
help="Hostname or ip address of the elasticsearch instance to connect to. "
|
||
|
|
"Can be set in env with 'ELASTIC_HOST'")
|
||
|
|
parser.add_argument('--elastic_port', default=cls.integer('ELASTIC_PORT', 9200), type=int,
|
||
|
|
help="Elasticsearch port to connect to. Can be set in env with 'ELASTIC_PORT'")
|
||
|
|
parser.add_argument('--elastic_notifier_host', default=cls.default('ELASTIC_NOTIFIER_HOST', 'localhost'),
|
||
|
|
type=str, help='elasticsearch sync notifier host, defaults to localhost')
|
||
|
|
parser.add_argument('--elastic_notifier_port', default=cls.integer('ELASTIC_NOTIFIER_PORT', 19080), type=int,
|
||
|
|
help='elasticsearch sync notifier port')
|
||
|
|
parser.add_argument('--es_index_prefix', default=cls.default('ES_INDEX_PREFIX', ''), type=str)
|
||
|
|
|
||
|
|
parser.add_argument('--query_timeout_ms', type=int, default=cls.integer('QUERY_TIMEOUT_MS', 10000),
|
||
|
|
help="Elasticsearch query timeout, in ms. Can be set in env with 'QUERY_TIMEOUT_MS'")
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_arg_parser(cls, args):
|
||
|
|
return cls(
|
||
|
|
db_dir=args.db_dir, elastic_host=args.elastic_host,
|
||
|
|
elastic_port=args.elastic_port, max_query_workers=args.max_query_workers, chain=args.chain,
|
||
|
|
es_index_prefix=args.es_index_prefix, reorg_limit=args.reorg_limit,
|
||
|
|
prometheus_port=args.prometheus_port, cache_all_tx_hashes=args.cache_all_tx_hashes,
|
||
|
|
cache_all_claim_txos=args.cache_all_claim_txos, blocking_channel_ids=args.blocking_channel_ids,
|
||
|
|
filtering_channel_ids=args.filtering_channel_ids, elastic_notifier_host=args.elastic_notifier_host,
|
||
|
|
elastic_notifier_port=args.elastic_notifier_port
|
||
|
|
)
|