diff options
author | Chris St. Pierre <chris.a.st.pierre@gmail.com> | 2012-05-08 15:09:54 -0400 |
---|---|---|
committer | Chris St. Pierre <chris.a.st.pierre@gmail.com> | 2012-05-08 15:36:02 -0400 |
commit | c35347887bb3d452d6104b13308d853b3da44b68 (patch) | |
tree | 569d5a6d94bfbd598a645c5511818d94b384acb4 /src/lib/Bcfg2/Server/Plugins/Cfg/CfgGenshiGenerator.py | |
parent | 16b7ac3cb7f9a49e6b7985059c3d08e984cf468c (diff) | |
download | bcfg2-c35347887bb3d452d6104b13308d853b3da44b68.tar.gz bcfg2-c35347887bb3d452d6104b13308d853b3da44b68.tar.bz2 bcfg2-c35347887bb3d452d6104b13308d853b3da44b68.zip |
modularized Cfg
Diffstat (limited to 'src/lib/Bcfg2/Server/Plugins/Cfg/CfgGenshiGenerator.py')
-rw-r--r-- | src/lib/Bcfg2/Server/Plugins/Cfg/CfgGenshiGenerator.py | 63 |
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) + |