first release
This commit is contained in:
parent
f8efc58597
commit
1f2edeea5a
1 changed files with 80 additions and 17 deletions
|
@ -8,41 +8,104 @@ class ConfigParser:
|
||||||
return dict_to_return
|
return dict_to_return
|
||||||
|
|
||||||
def add_section(self, section):
|
def add_section(self, section):
|
||||||
pass
|
self.config_dict[section] = {}
|
||||||
|
|
||||||
def has_section(self, section):
|
def has_section(self, section):
|
||||||
pass
|
if section in self.config_dict.keys():
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
|
||||||
def options(self, section):
|
def options(self, section):
|
||||||
pass
|
if not section in self.config_dict:
|
||||||
|
raise
|
||||||
|
return self.config_dict[section].keys()
|
||||||
|
|
||||||
def read(self, filename = None, fp = None):
|
def read(self, filename=None, fp=None):
|
||||||
if not fp:
|
if not fp and not filename:
|
||||||
|
print("ERROR : no filename and no fp")
|
||||||
|
raise
|
||||||
|
elif not fp and filename:
|
||||||
fp = open(filename)
|
fp = open(filename)
|
||||||
self.config_dict = {line.replace('[','').replace(']',''):{} for line in fp.read()\
|
|
||||||
if line.startswith('[') and line.endwith(']')
|
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):
|
def get(self, section, option):
|
||||||
|
if not self.has_section(section) \
|
||||||
def items(self, section):
|
or not self.has_option(section,option):
|
||||||
|
raise
|
||||||
def getboolean(self, section, option):
|
return self.config_dict[section][option]
|
||||||
|
|
||||||
def has_option(self, section, option):
|
def has_option(self, section, option):
|
||||||
"""Check for the existence of a given option in a given section."""
|
"""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 set(self, section, option, value=None):
|
def write(self, filename = None, fp = None):
|
||||||
"""Set an option."""
|
|
||||||
|
|
||||||
def write(self, fp):
|
|
||||||
"""Write an .ini-format representation of the configuration state."""
|
"""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):
|
def remove_option(self, section, option):
|
||||||
"""Remove an 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):
|
def remove_section(self, section):
|
||||||
"""Remove a file section."""
|
"""Remove a file section."""
|
||||||
|
if not self.has_section(section):
|
||||||
|
raise
|
||||||
|
del self.config_dict[section]
|
||||||
|
|
Loading…
Reference in a new issue