+ def create(self, filename):
+ if not os.path.exists(os.path.dirname(filename)):
+ os.makedirs(os.path.dirname(filename))
+ configfile = open(filename, 'w')
+ configfile.write(default_config)
+ configfile.close()
+
+ def load(self, filename):
+ if filename:
+ try:
+ self.config.read(filename)
+ except ConfigParser.MissingSectionHeaderError:
+ if filename.endswith('.xml'):
+ self.load_xml(filename)
+ else:
+ self.load_shell(filename)
+ self._files.append(filename)
+ self.set_attributes()
+
+ def load_xml(self, filename):
+ xml = XML(filename)
+ categories = xml.xpath('//configuration/variables/category')
+ for category in categories:
+ section_name = category.get('id')
+ if not self.config.has_section(section_name):
+ self.config.add_section(section_name)
+ options = category.xpath('./variablelist/variable')
+ for option in options:
+ option_name = option.get('id')
+ value = option.xpath('./value')[0].text
+ if not value:
+ value = ""
+ self.config.set(section_name, option_name, value)
+
+ def load_shell(self, filename):
+ f = open(filename, 'r')
+ for line in f:
+ try:
+ if line.startswith('#'):
+ continue
+ parts = line.strip().split("=")
+ if len(parts) < 2:
+ continue
+ option = parts[0]
+ value = parts[1].replace('"', '').replace("'", "")
+ section, var = self.locate_varname(option, strict=False)
+ if section and var:
+ self.set(section, var, value)
+ except:
+ pass
+ f.close()
+
+ def locate_varname(self, varname, strict=True):
+ varname = varname.lower()
+ sections = self.config.sections()
+ section_name = ""
+ var_name = ""
+ for section in sections:
+ if varname.startswith(section.lower()) and len(section) > len(section_name):
+ section_name = section.lower()
+ var_name = varname.replace(section_name, "")[1:]
+ if strict and not self.config.has_option(section_name, var_name):
+ raise ConfigParser.NoOptionError(var_name, section_name)
+ return (section_name, var_name)
+
+ def set_attributes(self):
+ sections = self.config.sections()
+ for section in sections:
+ for item in self.config.items(section):
+ name = "%s_%s" % (section, item[0])
+ value = item[1]
+ if isbool(value):
+ value = str2bool(value)
+ elif value.isdigit():
+ value = int(value)
+ setattr(self, name, value)
+ setattr(self, name.upper(), value)
+
+ def variables(self):
+ """
+ Return all variables.
+
+ Returns:
+
+ variables = { 'category_id': (category, variablelist) }
+
+ category = { 'id': "category_identifier",
+ 'name': "Category name",
+ 'description': "Category description" }
+
+ variablelist = { 'variable_id': variable }
+
+ variable = { 'id': "variable_identifier",
+ 'type': "variable_type",
+ 'value': "variable_value",
+ 'name': "Variable name",
+ 'description': "Variable description" }
+ """
+
+ variables = {}
+ for section in self.config.sections():
+ category = {
+ 'id': section,
+ 'name': section,
+ 'description': section,
+ }
+ variable_list = {}
+ for item in self.config.items(section):
+ var_name = item[0]
+ name = "%s_%s" % (section, var_name)
+ value = item[1]
+ if isbool(value):
+ value_type = bool
+ elif value.isdigit():
+ value_type = int
+ else:
+ value_type = str
+ variable = {
+ 'id': var_name,
+ 'type': value_type,
+ 'value': value,
+ 'name': name,
+ 'description': name,
+ }
+ variable_list[name] = variable
+ variables[section] = (category, variable_list)
+ return variables