8 from StringIO import StringIO
9 from sfa.util.xml import XML
16 return v.lower() in ("true", "false")
19 return v.lower() in ("true", "1")
23 def __init__(self, config_file='/etc/sfa/sfa_config'):
25 self.config_path = os.path.dirname(config_file)
26 self.config = ConfigParser.ConfigParser()
27 self.filename = config_file
28 if not os.path.isfile(self.filename):
29 self.create(self.filename)
30 self.load(self.filename)
35 DO NOT EDIT. This file was automatically generated at
39 """ % (time.asctime(), os.linesep.join(self._files))
41 # Get rid of the surrounding newlines
42 return header.strip().split(os.linesep)
44 def create(self, filename):
45 if not os.path.exists(os.path.dirname(filename)):
46 os.makedirs(os.path.dirname(filename))
47 configfile = open(filename, 'w')
48 configfile.write(default_config)
52 def load(self, filename):
55 self.config.read(filename)
56 except ConfigParser.MissingSectionHeaderError:
57 if filename.endswith('.xml'):
58 self.load_xml(filename)
60 self.load_shell(filename)
61 self._files.append(filename)
64 def load_xml(self, filename):
66 categories = xml.xpath('//configuration/variables/category')
67 for category in categories:
68 section_name = category.get('id')
69 if not self.config.has_section(section_name):
70 self.config.add_section(section_name)
71 options = category.xpath('./variablelist/variable')
72 for option in options:
73 option_name = option.get('id')
74 value = option.xpath('./value')[0].text
77 self.config.set(section_name, option_name, value)
79 def load_shell(self, filename):
80 f = open(filename, 'r')
83 if line.startswith('#'):
85 parts = line.strip().split("=")
89 value = parts[1].replace('"', '').replace("'","")
90 section, var = self.locate_varname(option, strict=False)
92 self.set(section, var, value)
97 def locate_varname(self, varname, strict=True):
98 varname = varname.lower()
99 sections = self.config.sections()
102 for section in sections:
103 if varname.startswith(section.lower()) and len(section) > len(section_name):
104 section_name = section.lower()
105 var_name = varname.replace(section_name, "")[1:]
106 if strict and not self.config.has_option(section_name, var_name):
107 raise ConfigParser.NoOptionError(var_name, section_name)
108 return (section_name, var_name)
110 def set_attributes(self):
111 sections = self.config.sections()
112 for section in sections:
113 for item in self.config.items(section):
114 name = "%s_%s" % (section, item[0])
117 value = str2bool(value)
118 elif value.isdigit():
120 setattr(self, name, value)
121 setattr(self, name.upper(), value)
125 Return all variables.
129 variables = { 'category_id': (category, variablelist) }
131 category = { 'id': "category_identifier",
132 'name': "Category name",
133 'description': "Category description" }
135 variablelist = { 'variable_id': variable }
137 variable = { 'id': "variable_identifier",
138 'type': "variable_type",
139 'value': "variable_value",
140 'name': "Variable name",
141 'description': "Variable description" }
145 for section in self.config.sections():
149 'description': section,
152 for item in self.config.items(section):
154 name = "%s_%s" % (section, var_name)
158 elif value.isdigit():
169 variable_list[name] = variable
170 variables[section] = (category, variable_list)
173 def verify(self, config1, config2, validate_method):
176 def validate_type(self, var_type, value):
180 def is_xml(config_file):
188 def is_ini(config_file):
190 c = ConfigParser.ConfigParser()
193 except ConfigParser.MissingSectionHeaderError:
197 def dump(self, sections = []):
198 sys.stdout.write(output_python())
200 def output_python(self, encoding = "utf-8"):
201 buf = codecs.lookup(encoding)[3](StringIO())
202 buf.writelines(["# " + line + os.linesep for line in self._header()])
204 for section in self.sections():
205 buf.write("[%s]%s" % (section, os.linesep))
206 for (name,value) in self.items(section):
207 buf.write("%s=%s%s" % (name,value,os.linesep))
208 buf.write(os.linesep)
209 return buf.getvalue()
211 def output_shell(self, show_comments = True, encoding = "utf-8"):
213 Return variables as a shell script.
216 buf = codecs.lookup(encoding)[3](StringIO())
217 buf.writelines(["# " + line + os.linesep for line in self._header()])
219 for section in self.sections():
220 for (name,value) in self.items(section):
221 # bash does not have the concept of NULL
223 option = "%s_%s" % (section.upper(), name.upper())
225 value = str(str2bool(value))
226 elif not value.isdigit():
227 value = '"%s"' % value
228 buf.write(option + "=" + value + os.linesep)
229 return buf.getvalue()
231 def output_php(selfi, encoding = "utf-8"):
233 Return variables as a PHP script.
236 buf = codecs.lookup(encoding)[3](StringIO())
237 buf.write("<?php" + os.linesep)
238 buf.writelines(["// " + line + os.linesep for line in self._header()])
240 for section in self.sections():
241 for (name,value) in self.items(section):
242 option = "%s_%s" % (section, name)
243 buf.write(os.linesep)
244 buf.write("// " + option + os.linesep)
247 buf.write("define('%s', %s);" % (option, value) + os.linesep)
249 buf.write("?>" + os.linesep)
251 return buf.getvalue()
253 def output_xml(self, encoding = "utf-8"):
256 def output_variables(self, encoding="utf-8"):
258 Return list of all variable names.
261 buf = codecs.lookup(encoding)[3](StringIO())
262 for section in self.sections():
263 for (name,value) in self.items(section):
264 option = "%s_%s" % (section,name)
265 buf.write(option + os.linesep)
267 return buf.getvalue()
270 def write(self, filename=None):
272 filename = self.filename
273 configfile = open(filename, 'w')
274 self.config.write(configfile)
276 def save(self, filename=None):
280 def get_trustedroots_dir(self):
281 return self.config_path + os.sep + 'trusted_roots'
283 def get_openflow_aggrMgr_info(self):
284 aggr_mgr_ip = 'localhost'
285 if (hasattr(self,'openflow_aggregate_manager_ip')):
286 aggr_mgr_ip = self.OPENFLOW_AGGREGATE_MANAGER_IP
289 if (hasattr(self,'openflow_aggregate_manager_port')):
290 aggr_mgr_port = self.OPENFLOW_AGGREGATE_MANAGER_PORT
292 return (aggr_mgr_ip,aggr_mgr_port)
294 def get_interface_hrn(self):
295 if (hasattr(self,'sfa_interface_hrn')):
296 return self.SFA_INTERFACE_HRN
300 def __getattr__(self, attr):
301 return getattr(self.config, attr)
303 if __name__ == '__main__':
305 if len(sys.argv) > 1:
306 filename = sys.argv[1]
307 config = Config(filename)