1 files changed, 139 insertions, 0 deletions
diff --git a/python/skytools/config.py b/python/skytools/config.py
new file mode 100644
index 00000000..de420322
--- /dev/null
+++ b/python/skytools/config.py
@@ -0,0 +1,139 @@
+
+"""Nicer config class."""
+
+import sys, os, ConfigParser, socket
+
+__all__ = ['Config']
+
+class Config(object):
+    """Bit improved ConfigParser.
+
+    Additional features:
+     - Remembers section.
+     - Acceps defaults in get() functions.
+     - List value support.
+    """
+    def __init__(self, main_section, filename, sane_config = 1):
+        """Initialize Config and read from file.
+
+        @param sane_config:  chooses between ConfigParser/SafeConfigParser.
+        """
+        defs = {
+            'job_name': main_section,
+            'service_name': main_section,
+            'host_name': socket.gethostname(),
+        }
+        if not os.path.isfile(filename):
+            raise Exception('Config file not found: '+filename)
+
+        self.filename = filename
+        self.sane_config = sane_config
+        if sane_config:
+            self.cf = ConfigParser.SafeConfigParser(defs)
+        else:
+            self.cf = ConfigParser.ConfigParser(defs)
+        self.cf.read(filename)
+        self.main_section = main_section
+        if not self.cf.has_section(main_section):
+            raise Exception("Wrong config file, no section '%s'"%main_section)
+
+    def reload(self):
+        """Re-reads config file."""
+        self.cf.read(self.filename)
+
+    def get(self, key, default=None):
+        """Reads string value, if not set then default."""
+        try:
+            return self.cf.get(self.main_section, key)
+        except ConfigParser.NoOptionError, det:
+            if default == None:
+                raise Exception("Config value not set: " + key)
+            return default
+
+    def getint(self, key, default=None):
+        """Reads int value, if not set then default."""
+        try:
+            return self.cf.getint(self.main_section, key)
+        except ConfigParser.NoOptionError, det:
+            if default == None:
+                raise Exception("Config value not set: " + key)
+            return default
+
+    def getboolean(self, key, default=None):
+        """Reads boolean value, if not set then default."""
+        try:
+            return self.cf.getboolean(self.main_section, key)
+        except ConfigParser.NoOptionError, det:
+            if default == None:
+                raise Exception("Config value not set: " + key)
+            return default
+
+    def getfloat(self, key, default=None):
+        """Reads float value, if not set then default."""
+        try:
+            return self.cf.getfloat(self.main_section, key)
+        except ConfigParser.NoOptionError, det:
+            if default == None:
+                raise Exception("Config value not set: " + key)
+            return default
+
+    def getlist(self, key, default=None):
+        """Reads comma-separated list from key."""
+        try:
+            s = self.cf.get(self.main_section, key).strip()
+            res = []
+            if not s:
+                return res
+            for v in s.split(","):
+                res.append(v.strip())
+            return res
+        except ConfigParser.NoOptionError, det:
+            if default == None:
+                raise Exception("Config value not set: " + key)
+            return default
+
+    def getfile(self, key, default=None):
+        """Reads filename from config.
+        
+        In addition to reading string value, expands ~ to user directory.
+        """
+        fn = self.get(key, default)
+        if fn == "" or fn == "-":
+            return fn
+        # simulate that the cwd is script location
+        #path = os.path.dirname(sys.argv[0])
+        #  seems bad idea, cwd should be cwd
+
+        fn = os.path.expanduser(fn)
+
+        return fn
+
+    def get_wildcard(self, key, values=[], default=None):
+        """Reads a wildcard property from conf and returns its string value, if not set then default."""
+        
+        orig_key = key
+        keys = [key]
+        
+        for wild in values:
+            key = key.replace('*', wild, 1)
+            keys.append(key)
+        keys.reverse()
+
+        for key in keys:
+            try:
+                return self.cf.get(self.main_section, key)
+            except ConfigParser.NoOptionError, det:
+                pass
+
+        if default == None:
+            raise Exception("Config value not set: " + orig_key)
+        return default
+    
+    def sections(self):
+        """Returns list of sections in config file, excluding DEFAULT."""
+        return self.cf.sections()
+
+    def clone(self, main_section):
+        """Return new Config() instance with new main section on same config file."""
+        return Config(main_section, self.filename, self.sane_config)
+