[rhmessaging-commits] rhmessaging commits: r2558 - mgmt/trunk/parsley/python/parsley.

[rhmessaging-commits at lists.jboss.org]

Author: justi9
Date: 2008-10-01 11:11:54 -0400 (Wed, 01 Oct 2008)
New Revision: 2558

Added:
   mgmt/trunk/parsley/python/parsley/config.py
Log:
Add a common config api to parsley

Added: mgmt/trunk/parsley/python/parsley/config.py
===================================================================
--- mgmt/trunk/parsley/python/parsley/config.py	                        (rev 0)
+++ mgmt/trunk/parsley/python/parsley/config.py	2008-10-01 15:11:54 UTC (rev 2558)
@@ -0,0 +1,76 @@
+import logging
+
+from ConfigParser import SafeConfigParser
+
+log = logging.getLogger("parsley.config")
+
+class Config(object):
+    def __init__(self):
+        self._params = list()
+        self._params_by_name = dict()
+
+    def init(self):
+        for param in self._params:
+            param.init()
+
+    def load_file(self, file):
+        conf = SafeConfigParser()
+        found = conf.read(file)
+
+        if found:
+            log.info("Read config file '%s'" % file)
+        else:
+            log.info("Config file '%s' not found" % file)
+
+        params = dict()
+
+        if (conf.has_section("main")): 
+            for key, value in conf.items("main"):
+                params[key] = value
+
+        self.load_dict(params)
+
+    def load_dict(self, params):
+        for sname, svalue in params.items():
+            param = self._params_by_name.get(sname)
+
+            if param:
+                name = param.name.replace("-", "_")
+                value = param.unmarshal(svalue)
+
+                setattr(self, name, value)
+
+            else:
+                log.info("Ignoring unrecognized parameter '%s'" % sname)
+
+    def prt(self):
+        print "Configuration:"
+
+        for param in self._params:
+            value = getattr(self, param.name.replace("-", "_"))
+
+            if value == param.default:
+                flag = " [default]"
+            else:
+                flag = ""
+
+            print "    %s = %s%s" % (param.name, value, flag)
+
+class ConfigParameter(object):
+    def __init__(self, config, name, type):
+        self.config = config
+        self.name = name
+        self.type = type
+        self.default = None
+
+        self.config._params.append(self)
+        self.config._params_by_name[self.name] = self
+
+    def init(self):
+        if hasattr(self.config, self.name):
+            raise Exception("Parameter '%s' already present" % self.name)
+
+        setattr(self.config, self.name.replace("-", "_"), self.default)
+
+    def unmarshal(self, string):
+        return self.type(string)