extras-buildsys/common BaseConfig.py,1.1,1.2

[Daniel Williams (dcbw)]

Author: dcbw

Update of /cvs/fedora/extras-buildsys/common
In directory cvs-int.fedora.redhat.com:/tmp/cvs-serv31571/common

Added Files:
	BaseConfig.py 
Log Message:
2005-08-24  Dan Williams <dcbw at redhat.com>

    * common/BaseConfig.py
      client/client.py
        - Rewrite the client to make the code less stupid,
            more easily followed, and to validate more
            input options
        - Also fixes RH #166692




Index: BaseConfig.py
===================================================================
RCS file: BaseConfig.py
diff -N BaseConfig.py
--- /dev/null	1 Jan 1970 00:00:00 -0000
+++ BaseConfig.py	24 Aug 2005 19:27:54 -0000	1.2
@@ -0,0 +1,92 @@
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Library General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+#
+# Copyright 2005 Dan Williams <dcbw at redhat.com> and Red Hat, Inc.
+
+import os
+from ConfigParser import ConfigParser
+
+
+class ConfigError(Exception):
+    pass
+
+class BaseConfig:
+    def __init__(self, filename):
+        self._filename = filename
+        self._config = ConfigParser()
+
+    def open(self, filename=None):
+        if not filename:
+            filename = self._filename
+        if not os.path.exists(filename):
+            raise ConfigError("Config file '%s' missing." % filename)
+        self._config.read(filename)
+        self._filename = filename
+
+    def has_option(self, section, name):
+        return self._config.has_option(section, name)
+
+    def get_option(self, section, name):
+        if not self._config.has_section(section):
+            raise ConfigError("Invalid section: %s" % section)
+        if self._config.has_option(section, name):
+            return self._config.get(section, name)
+        raise ConfigError("Config file %s does not have option: %s/%s" % (self._filename, section, name))
+
+    def get_str(self, section, name):
+        return self.get_option(section, name)
+
+    def get_bool(self, section, name):
+        opt = self.get_option(section, name)
+        if type(opt) == type(""):
+            if opt.lower() == 'yes' or opt.lower() == 'true':
+                return True
+            elif opt.lower() == 'no' or opt.lower() == 'false':
+                return False
+        raise ConfigError("Invalid format for %s/%s.  Should be one of [yes, no, true, false]." % (section, name))
+
+    def get_list(self, section, name):
+        opt = self.get_option(section, name)
+        if type(opt) == type(""):
+            if not len(opt):
+                return []
+            try:
+                return opt.split()
+            except Exception:
+                pass
+        raise ConfigError("Invalid format for %s/%s.  Should be a space-separate list." % (section, name))
+
+    def get_int(self, section, name):
+        opt = self.get_option(section, name)
+        try:
+            return int(opt)
+        except Exception:
+            pass
+        raise ConfigError("Invalid format for %s/%s.  Should be a valid integer." % (section, name))
+
+    def add_section(self, section):
+        self._config.add_section(section)
+
+    def set_option(self, section, name, value):
+        if not self._config.has_section(section):
+            self._config.add_section(section)
+        self._config.set(section, name, value)
+
+    def save(self, filename=None):
+        if not filename:
+            filename = self._filename
+        fp = open(filename, 'w')
+        self._config.write(fp)
+        self._filename = filename
+