def get_rest_cacher_dir():
cacher_dir = None
if config_has_section('rest') and config_has_option('rest', 'cacher_dir'):
cacher_dir = config_get('rest', 'cacher_dir')
if cacher_dir and os.path.exists(cacher_dir):
return cacher_dir
raise Exception("cacher_dir is not defined or it doesn't exist")
def setup_logging(name):
"""
Setup logging
"""
if config_has_section('common') and config_has_option('common', 'loglevel'):
loglevel = getattr(logging, config_get('common', 'loglevel').upper())
else:
loglevel = logging.INFO
if config_has_section('common') and config_has_option('common', 'logdir'):
logging.basicConfig(filename=os.path.join(config_get('common', 'logdir'), name),
level=loglevel,
format='%(asctime)s\t%(name)s\t%(levelname)s\t%(message)s')
else:
logging.basicConfig(stream=sys.stdout, level=loglevel,
format='%(asctime)s\t%(name)s\t%(levelname)s\t%(message)s')
def load_config_agents():
if config_has_section(Sections.Main) and config_has_option(
Sections.Main, 'agents'):
agents = config_get(Sections.Main, 'agents')
agents = agents.split(',')
agents = [d.strip() for d in agents]
return agents
return []
def load_plugin_sequence(config_section, config_option='plugin_sequence'):
"""
load plugin sequence
"""
plugin_sequence = []
if config_has_section(config_section) and config_has_option(
config_section, config_option):
plugin_sequence = config_get(config_section, config_option)
plugin_sequence = plugin_sequence.split(",")
plugin_sequence = [plugin.strip() for plugin in plugin_sequence]
return plugin_sequence
def get_rest_url_prefix():
if config_has_section('rest') and config_has_option('rest', 'url_prefix'):
url_prefix = config_get('rest', 'url_prefix')
else:
url_prefix = None
if url_prefix:
while url_prefix.startswith('/'):
url_prefix = url_prefix[1:]
while url_prefix.endswith('/'):
url_prefix = url_prefix[:-1]
url_prefix = '/' + url_prefix
return url_prefix
def load_plugins(config_section):
"""
Load plugins
"""
plugins = {}
if config_has_section(config_section):
options = config_list_options(config_section)
for option, value in options:
if option.startswith('plugin.'):
if option.count('.') == 1:
plugin_name = option.replace('plugin.', '').strip()
plugins[plugin_name] = load_plugin(config_section,
plugin_name, value)
return plugins
def load_agent_attrs(section):
"""
Load agent attributes
"""
attrs = {}
logging.info("Loading config for section: %s" % section)
if config_has_section(section):
options = config_list_options(section)
for option, value in options:
if not option.startswith('plugin.'):
if isinstance(value, str) and value.lower() == 'true':
value = True
if isinstance(value, str) and value.lower() == 'false':
value = False
attrs[option] = value
return attrs
def load_plugin_attributes(config_section, name, plugin):
"""
Load plugin attributes
"""
attrs = {}
if config_has_section(config_section):
options = config_list_options(config_section)
for option, value in options:
plugin_prefix = 'plugin.%s.' % name
if option.startswith(plugin_prefix):
attr_name = option.replace(plugin_prefix, '')
if isinstance(value, str) and value.lower() == 'true':
value = True
if isinstance(value, str) and value.lower() == 'false':
value = False
attrs[attr_name] = value
return attrs
def get_rest_debug():
if config_has_section('rest') and config_has_option('rest', 'debug'):
return config_get_bool('rest', 'debug')
return False
def get_cache_url():
if config_has_section('cache') and config_has_option('cache', 'url'):
return config_get('cache', 'url')
return '127.0.0.1:11211'
