-#!/usr/bin/python
+#!/usr/bin/env python3
+
+# pylint: disable=c0111, c0103, r0904, r1705, r0201
+
import sys
import os
import time
-import ConfigParser
-import tempfile
-import codecs
-from StringIO import StringIO
+#import tempfile
+from io import StringIO
from sfa.util.xml import XML
+import configparser
default_config = \
+ """
"""
-"""
-def isbool(v):
- return v.lower() in ("true", "false")
+
+def isbool(val):
+ return val.lower() in ("true", "false")
+
def str2bool(v):
- return v.lower() in ("true", "1")
+ return v.lower() in ("true", "1")
+
class Config:
-
+
def __init__(self, config_file='/etc/sfa/sfa_config'):
self._files = []
self.config_path = os.path.dirname(config_file)
- self.config = ConfigParser.ConfigParser()
+ self.config = configparser.ConfigParser()
self.filename = config_file
if not os.path.isfile(self.filename):
self.create(self.filename)
self.load(self.filename)
-
def _header(self):
header = """
configfile = open(filename, 'w')
configfile.write(default_config)
configfile.close()
-
def load(self, filename):
if filename:
try:
self.config.read(filename)
- except ConfigParser.MissingSectionHeaderError:
+ except configparser.MissingSectionHeaderError:
if filename.endswith('.xml'):
self.load_xml(filename)
else:
if not value:
value = ""
self.config.set(section_name, option_name, value)
-
+
def load_shell(self, filename):
f = open(filename, 'r')
for line in f:
if len(parts) < 2:
continue
option = parts[0]
- value = parts[1].replace('"', '').replace("'","")
+ 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()
+ f.close()
def locate_varname(self, varname, strict=True):
varname = varname.lower()
section_name = section.lower()
var_name = varname.replace(section_name, "")[1:]
if strict and not self.config.has_option(section_name, var_name):
- raise ConfigParser.NoOptionError(var_name, section_name)
- return (section_name, var_name)
+ raise configparser.NoOptionError(var_name, section_name)
+ return (section_name, var_name)
def set_attributes(self):
sections = self.config.sections()
if isbool(value):
value = str2bool(value)
elif value.isdigit():
- value = int(value)
+ value = int(value)
setattr(self, name, value)
setattr(self, name.upper(), value)
}
variable_list = {}
for item in self.config.items(section):
- var_name = item[0]
+ var_name = item[0]
name = "%s_%s" % (section, var_name)
value = item[1]
if isbool(value):
}
variable_list[name] = variable
variables[section] = (category, variable_list)
- return variables
+ return variables
- def verify(self, config1, config2, validate_method):
+ def verify(self, _config1, _config2, _validate_method):
return True
- def validate_type(self, var_type, value):
+ def validate_type(self, _var_type, _value):
return True
@staticmethod
def is_xml(config_file):
try:
- x = Xml(config_file)
- return True
+ _x = Xml(config_file)
+ return True
except:
return False
@staticmethod
def is_ini(config_file):
try:
- c = ConfigParser.ConfigParser()
+ c = configparser.ConfigParser()
c.read(config_file)
return True
- except ConfigParser.MissingSectionHeaderError:
+ except configparser.MissingSectionHeaderError:
return False
+ def dump(self, sections=None):
+ if sections is None:
+ sections = []
+ sys.stdout.write(self.output_python())
- def dump(self, sections = []):
- sys.stdout.write(output_python())
+ def output_python(self):
+ buf = StringIO()
+ buf.writelines(["# " + line + os.linesep for line in self._header()])
- def output_python(self, encoding = "utf-8"):
- buf = codecs.lookup(encoding)[3](StringIO())
- buf.writelines(["# " + line + os.linesep for line in self._header()])
-
for section in self.sections():
buf.write("[%s]%s" % (section, os.linesep))
- for (name,value) in self.items(section):
- buf.write("%s=%s%s" % (name,value,os.linesep))
+ for (name, value) in self.items(section):
+ buf.write("%s=%s%s" % (name, value, os.linesep))
buf.write(os.linesep)
return buf.getvalue()
-
- def output_shell(self, show_comments = True, encoding = "utf-8"):
+
+ def output_shell(self):
"""
Return variables as a shell script.
"""
- buf = codecs.lookup(encoding)[3](StringIO())
+ buf = StringIO()
buf.writelines(["# " + line + os.linesep for line in self._header()])
for section in self.sections():
- for (name,value) in self.items(section):
+ for (name, value) in self.items(section):
# bash does not have the concept of NULL
if value:
option = "%s_%s" % (section.upper(), name.upper())
if isbool(value):
value = str(str2bool(value))
elif not value.isdigit():
- value = '"%s"' % value
+ value = '"%s"' % value
buf.write(option + "=" + value + os.linesep)
- return buf.getvalue()
+ return buf.getvalue()
- def output_php(selfi, encoding = "utf-8"):
+ def output_php(selfi):
"""
Return variables as a PHP script.
"""
- buf = codecs.lookup(encoding)[3](StringIO())
+ buf = StringIO()
buf.write("<?php" + os.linesep)
buf.writelines(["// " + line + os.linesep for line in self._header()])
for section in self.sections():
- for (name,value) in self.items(section):
+ for (name, value) in self.items(section):
option = "%s_%s" % (section, name)
buf.write(os.linesep)
buf.write("// " + option + os.linesep)
buf.write("?>" + os.linesep)
- return buf.getvalue()
+ return buf.getvalue()
- def output_xml(self, encoding = "utf-8"):
+ def output_xml(self):
pass
- def output_variables(self, encoding="utf-8"):
+ def output_variables(self):
"""
Return list of all variable names.
"""
- buf = codecs.lookup(encoding)[3](StringIO())
+ buf = StringIO()
for section in self.sections():
- for (name,value) in self.items(section):
- option = "%s_%s" % (section,name)
+ for (name, _value) in self.items(section):
+ option = "%s_%s" % (section, name)
buf.write(option + os.linesep)
return buf.getvalue()
- pass
-
+
def write(self, filename=None):
if not filename:
filename = self.filename
- configfile = open(filename, 'w')
+ configfile = open(filename, 'w')
self.config.write(configfile)
-
+
def save(self, filename=None):
self.write(filename)
-
def get_trustedroots_dir(self):
return self.config_path + os.sep + 'trusted_roots'
def get_openflow_aggrMgr_info(self):
aggr_mgr_ip = 'localhost'
- if (hasattr(self,'openflow_aggregate_manager_ip')):
+ if hasattr(self, 'openflow_aggregate_manager_ip'):
aggr_mgr_ip = self.OPENFLOW_AGGREGATE_MANAGER_IP
aggr_mgr_port = 2603
- if (hasattr(self,'openflow_aggregate_manager_port')):
+ if hasattr(self, 'openflow_aggregate_manager_port'):
aggr_mgr_port = self.OPENFLOW_AGGREGATE_MANAGER_PORT
- return (aggr_mgr_ip,aggr_mgr_port)
+ return (aggr_mgr_ip, aggr_mgr_port)
def get_interface_hrn(self):
- if (hasattr(self,'sfa_interface_hrn')):
+ if hasattr(self, 'sfa_interface_hrn'):
return self.SFA_INTERFACE_HRN
else:
return "plc"
return getattr(self.config, attr)
if __name__ == '__main__':
- filename = None
- if len(sys.argv) > 1:
- filename = sys.argv[1]
- config = Config(filename)
- else:
- config = Config()
- config.dump()
-
+ def main():
+ filename = None
+ if len(sys.argv) > 1:
+ filename = sys.argv[1]
+ config = Config(filename)
+ else:
+ config = Config()
+ config.dump()
+ main()