utils/test/dashboard/dashboard/conf/config.py

   1 #! /usr/bin/env python
   2
   3 import urlparse
   4 from ConfigParser import SafeConfigParser, NoOptionError
   5
   6
   7 class ParseError(Exception):
   8     """
   9     Custom exception class for config file
  10     """
  11
  12     def __init__(self, message):
  13         self.msg = message
  14
  15     def __str__(self):
  16         return 'error parsing config file : %s' % self.msg
  17
  18
  19 class APIConfig:
  20     """
  21     The purpose of this class is to load values correctly from the config file.
  22     Each key is declared as an attribute in __init__() and linked in parse()
  23     """
  24
  25     def __init__(self):
  26         self._default_config_location = "/etc/dashboard/config.ini"
  27         self.es_url = 'http://localhost:9200'
  28         self.es_creds = None
  29         self.kibana_url = None
  30         self.js_path = None
  31         self.index_url = None
  32
  33     def _get_str_parameter(self, section, param):
  34         try:
  35             return self._parser.get(section, param)
  36         except NoOptionError:
  37             raise ParseError("[%s.%s] parameter not found" % (section, param))
  38
  39     def _get_int_parameter(self, section, param):
  40         try:
  41             return int(self._get_str_parameter(section, param))
  42         except ValueError:
  43             raise ParseError("[%s.%s] not an int" % (section, param))
  44
  45     def _get_bool_parameter(self, section, param):
  46         result = self._get_str_parameter(section, param)
  47         if str(result).lower() == 'true':
  48             return True
  49         if str(result).lower() == 'false':
  50             return False
  51
  52         raise ParseError(
  53             "[%s.%s : %s] not a boolean" % (section, param, result))
  54
  55     @staticmethod
  56     def parse(config_location=None):
  57         obj = APIConfig()
  58
  59         if config_location is None:
  60             config_location = obj._default_config_location
  61
  62         obj._parser = SafeConfigParser()
  63         obj._parser.read(config_location)
  64         if not obj._parser:
  65             raise ParseError("%s not found" % config_location)
  66
  67         # Linking attributes to keys from file with their sections
  68         obj.es_url = obj._get_str_parameter("elastic", "url")
  69         obj.es_creds = obj._get_str_parameter("elastic", "creds")
  70         obj.kibana_url = obj._get_str_parameter("kibana", "url")
  71         obj.js_path = obj._get_str_parameter("kibana", "js_path")
  72         index = obj._get_str_parameter("elastic", "index")
  73         obj.index_url = urlparse.urljoin(obj.es_url, index)
  74
  75         return obj
  76
  77     def __str__(self):
  78         return "elastic_url = %s \n" \
  79                "elastic_creds = %s \n" \
  80                "kibana_url = %s \n" \
  81                "index_url = %s \n" \
  82                "js_path = %s \n" % (self.es_url,
  83                                     self.es_creds,
  84                                     self.kibana_url,
  85                                     self.index_url,
  86                                     self.js_path)