diff options
Diffstat (limited to 'src/lib/Bcfg2/Server/Plugins/Cfg/CfgCheetahGenerator.py')
-rw-r--r-- | src/lib/Bcfg2/Server/Plugins/Cfg/CfgCheetahGenerator.py | 33 |
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..3edd1d8cb --- /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 ImportError: + 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() |