"""
"""
+def isbool(v):
+ return v.lower() in ("true", "false")
+
+def str2bool(v):
+ return v.lower() in ("true", "1")
+
class Config:
def __init__(self, config_file='/etc/sfa/sfa_config'):
def load(self, filename):
if filename:
- if filename.endswith('.xml'):
- try:
- self.load_xml(filename)
- except:
- self.config.read(filename)
- else:
+ try:
self.config.read(filename)
- self._files.append(filename)
- self.set_attributes()
-
+ 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')
value = ""
self.config.set(section_name, option_name, value)
-
- def locate_varname(self, varname):
+ 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 = ""
if varname.startswith(section.lower()) and len(section) > len(section_name):
section_name = section.lower()
var_name = varname.replace(section_name, "")[1:]
- if not self.config.has_option(section_name, var_name):
- raise ConfigParser.NoOptionError(varname, section_name)
+ 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):
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 validate_type(self, var_type, value):
return True
+ @staticmethod
+ def is_xml(config_file):
+ try:
+ x = Xml(config_file)
+ return True
+ except:
+ return False
+
+ @staticmethod
+ def is_ini(config_file):
+ try:
+ c = ConfigParser.ConfigParser()
+ c.read(config_file)
+ return True
+ except ConfigParser.MissingSectionHeaderError:
+ return False
+
+
def dump(self, sections = []):
sys.stdout.write(output_python())
# bash does not have the concept of NULL
if value:
option = "%s_%s" % (section.upper(), name.upper())
- value = "'%s'" % value
+ if isbool(value):
+ value = str(str2bool(value))
+ elif not value.isdigit():
+ value = '"%s"' % value
buf.write(option + "=" + value + os.linesep)
return buf.getvalue()
def output_xml(self, encoding = "utf-8"):
pass
- def output_variables(self):
+ def output_variables(self, encoding="utf-8"):
"""
Return list of all variable names.
"""