class ConfigParser: def __init__(self): self.config_dict = {} def sections(self): """Return a list of section names, excluding [DEFAULT]""" dict_to_return = [section for section in self.config_dict.keys() if not section in "DEFAULT"] return dict_to_return def add_section(self, section): self.config_dict[section] = {} def has_section(self, section): if section in self.config_dict.keys(): return True else: return False def options(self, section): if not section in self.config_dict: raise return self.config_dict[section].keys() def read(self, filename=None, fp=None): if not fp and not filename: print("ERROR : no filename and no fp") raise elif not fp and filename: fp = open(filename) content = fp.read() fp.close() self.config_dict = {line.replace('[','').replace(']',''):{} for line in content.split('\n')\ if line.startswith('[') and line.endswith(']') } striped_content = [line.strip() for line in content.split('\n')] for section in self.config_dict.keys(): start_index = striped_content.index('[%s]' % section) end_flag = [line for line in striped_content[start_index + 1:] if line.startswith('[')] if not end_flag: end_index = None else: end_index = striped_content.index(end_flag[0]) block = striped_content[start_index + 1 : end_index] options = [line.split('=')[0].strip() for line in block if '=' in line] for option in options: start_flag = [line for line in block if line.startswith(option) and '=' in line] start_index = block.index(start_flag[0]) end_flag = [line for line in block[start_index + 1:] if '=' in line] if not end_flag: end_index = None else: end_index = block.index(end_flag[0]) values = [value.split('=')[-1].strip() for value in block[start_index:end_index] if value] if not values: values = None elif len(values) == 1: values = values[0] self.config_dict[section][option] = values def get(self, section, option): if not self.has_section(section) \ or not self.has_option(section,option): raise return self.config_dict[section][option] def has_option(self, section, option): """Check for the existence of a given option in a given section.""" if not section in self.config_dict: raise if option in self.config_dict[section].keys(): return True else: return False def write(self, filename = None, fp = None): """Write an .ini-format representation of the configuration state.""" if not fp and not filename: print("ERROR : no filename and no fp") raise elif not fp and filename: fp = open(filename,'w') for section in self.config_dict.keys(): fp.write('[%s]\n' % section) for option in self.config_dict[section].keys(): fp.write('\n%s =' % option) values = self.config_dict[section][option] if type(values) == type([]): fp.write('\n ') values = '\n '.join(values) else: fp.write(' ') fp.write(values) fp.write('\n') fp.write('\n') def remove_option(self, section, option): """Remove an option.""" if not self.has_section(section) \ or not self.has_option(section,option): raise del self.config_dict[section][option] def remove_section(self, section): """Remove a file section.""" if not self.has_section(section): raise del self.config_dict[section]