1 ##############################################################################
2 # Copyright (c) 2015 Orange
3 # guyrodrigue.koffi@orange.com / koffirodrigue@gmail.com
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 # feng.xiaowei@zte.com.cn remove prepare_put_request 5-30-2016
9 ##############################################################################
14 class ParseError(Exception):
16 Custom exception class for config file
19 def __init__(self, message):
23 return 'error parsing config file : %s' % self.msg
26 class APIConfig(object):
28 The purpose of this class is to load values correctly from the config file.
29 Each key is declared as an attribute in __init__() and linked in parse()
33 self._set_default_config()
35 self.mongo_dbname = None
37 self.api_debug_on = None
38 self.api_authenticate_on = None
40 self.swagger_base_url = None
42 def _set_default_config(self):
43 venv = os.getenv('VIRTUAL_ENV')
44 self._default_config = os.path.join('/' if not venv else venv,
45 'etc/opnfv_testapi/config.ini')
47 def _get_parameter(self, section, param):
49 return self._parser.get(section, param)
50 except ConfigParser.NoOptionError:
51 raise ParseError("No parameter: [%s.%s]" % (section, param))
53 def _get_int_parameter(self, section, param):
55 return int(self._get_parameter(section, param))
57 raise ParseError("Not int: [%s.%s]" % (section, param))
59 def _get_bool_parameter(self, section, param):
60 result = self._get_parameter(section, param)
61 if str(result).lower() == 'true':
63 if str(result).lower() == 'false':
67 "Not boolean: [%s.%s : %s]" % (section, param, result))
70 def parse(config_location=None):
73 if config_location is None:
74 config_location = obj._default_config
76 if not os.path.exists(config_location):
77 raise ParseError("%s not found" % config_location)
79 obj._parser = ConfigParser.SafeConfigParser()
80 obj._parser.read(config_location)
82 # Linking attributes to keys from file with their sections
83 obj.mongo_url = obj._get_parameter("mongo", "url")
84 obj.mongo_dbname = obj._get_parameter("mongo", "dbname")
86 obj.api_port = obj._get_int_parameter("api", "port")
87 obj.api_debug_on = obj._get_bool_parameter("api", "debug")
88 obj.api_authenticate_on = obj._get_bool_parameter("api",
91 obj.swagger_base_url = obj._get_parameter("swagger", "base_url")