modularized Cfg

1 files changed, 33 insertions, 0 deletions

diff --git a/src/lib/Bcfg2/Server/Plugins/Cfg/CfgCheetahGenerator.py b/src/lib/Bcfg2/Server/Plugins/Cfg/CfgCheetahGenerator.py
new file mode 100644
index 000000000..08f01e005
--- /dev/null
+++ b/src/lib/Bcfg2/Server/Plugins/Cfg/CfgCheetahGenerator.py
@@ -0,0 +1,33 @@
+import copy
+import logging
+import Bcfg2.Server.Plugin
+from Bcfg2.Server.Plugins.Cfg import CfgGenerator
+
+logger = logging.getLogger(__name__)
+
+try:
+    import Cheetah.Template
+    import Cheetah.Parser
+    have_cheetah = True
+except:
+    have_cheetah = False
+
+
+class CfgCheetahGenerator(CfgGenerator):
+    __extensions__ = ['cheetah']
+    settings = dict(useStackFrames=False)
+
+    def __init__(self, fname, spec, encoding):
+        CfgGenerator.__init__(self, fname, spec, encoding)
+        if not have_cheetah:
+            msg = "Cfg: Cheetah is not available: %s" % entry.get("name")
+            logger.error(msg)
+            raise Bcfg2.Server.Plugin.PluginExecutionError(msg)
+
+    def get_data(self, entry, metadata):
+        template = Cheetah.Template.Template(self.data,
+                                             compilerSettings=self.settings)
+        template.metadata = metadata
+        template.path = entry.get('realname', entry.get('name'))
+        template.source_path = self.path
+        return template.respond()