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.
37 lines
1010 B
37 lines
1010 B
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)
|
|
|