X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Futil%2Fconfig.py;h=c25ec4422184867a69a690a4ca1e33ad2bbdfa50;hb=fb54f03086be76e9b5a2c34ec63a75c13752394b;hp=2c06736590079be4493ace48b2c4236821245912;hpb=cc6bcae9aaa675416b01af086c6ea93a91986464;p=sfa.git diff --git a/sfa/util/config.py b/sfa/util/config.py index 2c067365..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'): @@ -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 @@ -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()