modularized Cfg

1 files changed, 63 insertions, 0 deletions

diff --git a/src/lib/Bcfg2/Server/Plugins/Cfg/CfgGenshiGenerator.py b/src/lib/Bcfg2/Server/Plugins/Cfg/CfgGenshiGenerator.py
new file mode 100644
index 000000000..5e3b37127
--- /dev/null
+++ b/src/lib/Bcfg2/Server/Plugins/Cfg/CfgGenshiGenerator.py
@@ -0,0 +1,63 @@
+import sys
+import logging
+import Bcfg2.Server.Plugin
+from Bcfg2.Server.Plugins.Cfg import CfgGenerator
+
+logger = logging.getLogger(__name__)
+
+try:
+    import genshi.core
+    from genshi.template import TemplateLoader, NewTextTemplate
+    have_genshi = True
+except:
+    have_genshi = False
+
+# snipped from TGenshi
+def removecomment(stream):
+    """A genshi filter that removes comments from the stream."""
+    for kind, data, pos in stream:
+        if kind is genshi.core.COMMENT:
+            continue
+        yield kind, data, pos
+
+
+class CfgGenshiGenerator(CfgGenerator):
+    __extensions__ = ['genshi']
+
+    def __init__(self, fname, spec, encoding):
+        CfgGenerator.__init__(self, fname, spec, encoding)
+        self.loader = TemplateLoader()
+        if not have_genshi:
+            msg = "Cfg: Genshi is not available: %s" % entry.get("name")
+            logger.error(msg)
+            raise Bcfg2.Server.Plugin.PluginExecutionError(msg)
+
+    @classmethod
+    def ignore(cls, basename, event):
+        return (event.filename.endswith(".genshi_include") or
+                CfgGenerator.ignore(basename, event))
+
+    def get_data(self, entry, metadata):
+        fname = entry.get('realname', entry.get('name'))
+        stream = \
+            self.template.generate(name=fname,
+                                   metadata=metadata,
+                                   path=self.name).filter(removecomment)
+        try:
+            return stream.render('text', encoding=self.encoding,
+                                 strip_whitespace=False)
+        except TypeError:
+            return stream.render('text', encoding=self.encoding)
+
+    def handle_event(self, event):
+        if event.code2str() == 'deleted':
+            return
+        try:
+            self.template = self.loader.load(self.name, cls=NewTextTemplate,
+                                             encoding=self.encoding)
+        except Exception:
+            msg = "Cfg: Could not load template %s: %s" % (self.name,
+                                                           sys.exc_info()[1])
+            logger.error(msg)
+            raise Bcfg2.Server.Plugin.PluginExecutionError(msg)
+