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.
43 lines
1.3 KiB
43 lines
1.3 KiB
import os |
|
import json |
|
from dotenv import load_dotenv |
|
|
|
load_dotenv() |
|
|
|
PARENT_FOLDER = os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__)))) |
|
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) |
|
else: |
|
print('Config file %s doesn`t exist, using environment variables / defaults' % CONFIG_FILE) |
|
|
|
|
|
def get_config_field(field: str, default_val = None, allowed_values = []): |
|
value = None |
|
if field in os.environ and os.environ[field] != '': |
|
value = os.environ[field] |
|
elif config_exists and field in config and config[field] != '': |
|
value = config[field] |
|
elif default_val is not None: |
|
value = default_val |
|
|
|
if len(allowed_values) > 0 and value not in allowed_values: |
|
raise Exception('{} value must be one of: {}, got: {}'.format(field, allowed_values, value)) |
|
|
|
if value is None: |
|
raise Exception('Please configure {}'.format(field)) |
|
|
|
return value |
|
|
|
HASTIC_SERVER_URL = get_config_field('HASTIC_SERVER_URL', 'ws://localhost:8002') |
|
LOGGING_LEVEL = get_config_field( |
|
'HS_AN_LOGGING_LEVEL', |
|
'INFO', |
|
# TODO: make values case insensitive |
|
['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'] |
|
) |
|
LEARNING_TIMEOUT = get_config_field('LEARNING_TIMEOUT', 120)
|
|
|