X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Futil%2Fconfig.py;h=c25ec4422184867a69a690a4ca1e33ad2bbdfa50;hb=8ece6aaf557800371b7366e22af7a6ab04614d10;hp=6fdb48d275794d3856498a6a456324d4b9effb6c;hpb=2073b53dc78978a8d31e70f80895f1597a691023;p=sfa.git diff --git a/sfa/util/config.py b/sfa/util/config.py index 6fdb48d2..c25ec442 100644 --- a/sfa/util/config.py +++ b/sfa/util/config.py @@ -12,6 +12,12 @@ default_config = \ """ """ +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'): @@ -47,7 +53,7 @@ DO NOT EDIT. This file was automatically generated at if filename: try: self.config.read(filename) - except: + except ConfigParser.MissingSectionHeaderError: if filename.endswith('.xml'): self.load_xml(filename) else: @@ -107,9 +113,62 @@ DO NOT EDIT. This file was automatically generated at 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 def verify(self, config1, config2, validate_method): return True @@ -131,7 +190,7 @@ DO NOT EDIT. This file was automatically generated at c = ConfigParser.ConfigParser() c.read(config_file) return True - except: + except ConfigParser.MissingSectionHeaderError: return False @@ -162,8 +221,10 @@ DO NOT EDIT. This file was automatically generated at # bash does not have the concept of NULL if value: option = "%s_%s" % (section.upper(), name.upper()) - if not value.isdigit() and not bool(value): - 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()