You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
|
import os
|
|
|
|
|
import json
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
PARENT_FOLDER = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
|
|
|
|
|
DATA_FOLDER = os.path.join(PARENT_FOLDER, 'data')
|
|
|
|
|
CONFIG_FILE = os.path.join(PARENT_FOLDER, 'config.json')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config_exists = os.path.isfile(CONFIG_FILE)
|
|
|
|
|
if config_exists:
|
|
|
|
|
with open(CONFIG_FILE) as f:
|
|
|
|
|
config = json.load(f)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def get_config_field(field, default_val = None):
|
|
|
|
|
|
|
|
|
|
if field in os.environ:
|
|
|
|
|
return os.environ[field]
|
|
|
|
|
|
|
|
|
|
if config_exists and field in config:
|
|
|
|
|
return config[field]
|
|
|
|
|
|
|
|
|
|
if default_val is not None:
|
|
|
|
|
return default_val
|
|
|
|
|
|
|
|
|
|
raise Exception('Please configure {}'.format(field))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
DATASET_FOLDER = os.path.join(DATA_FOLDER, 'datasets')
|
|
|
|
|
MODELS_FOLDER = os.path.join(DATA_FOLDER, 'models')
|
|
|
|
|
METRICS_FOLDER = os.path.join(DATA_FOLDER, 'metrics')
|
|
|
|
|
|
|
|
|
|
HASTIC_API_KEY = get_config_field('HASTIC_API_KEY')
|
|
|
|
|
|
|
|
|
|
ZMQ_DEV_PORT = get_config_field('ZMQ_DEV_PORT', '8002')
|
|
|
|
|
ZMQ_CONNECTION_STRING = get_config_field('ZMQ_CONNECTION_STRING', 'tcp://*:%s' % ZMQ_DEV_PORT)
|
