From fd787917bdcd0aaf5f5b0b99dfa5e2a992d734a4 Mon Sep 17 00:00:00 2001 From: Narayan Desai Date: Wed, 18 Jul 2007 19:00:57 +0000 Subject: Initial TGenshi checking (from jcollie) git-svn-id: https://svn.mcs.anl.gov/repos/bcfg/trunk/bcfg2@3476 ce84e21b-d406-0410-9b95-82705330c041 --- src/lib/Server/Plugins/TGenshi.py | 145 +++++++++++++++++++++++++++++++++++++ src/lib/Server/Plugins/__init__.py | 3 +- 2 files changed, 147 insertions(+), 1 deletion(-) create mode 100644 src/lib/Server/Plugins/TGenshi.py (limited to 'src/lib/Server/Plugins') diff --git a/src/lib/Server/Plugins/TGenshi.py b/src/lib/Server/Plugins/TGenshi.py new file mode 100644 index 000000000..da6dcf956 --- /dev/null +++ b/src/lib/Server/Plugins/TGenshi.py @@ -0,0 +1,145 @@ +'''This module implements a templating generator based on Genshi''' +__revision__ = '$Revision$' + +from genshi.template import TemplateLoader, TextTemplate, MarkupTemplate, TemplateError +import logging, lxml.etree, posixpath, re, os +import Bcfg2.Server.Plugin + +logger = logging.getLogger('Bcfg2.Plugins.TGenshi') +info = re.compile('^owner:(\s)*(?P\w+)$|group:(\s)*(?P\w+)$|' + + 'perms:(\s)*(?P\w+)$') + +class TemplateFile: + '''Template file creates Genshi template structures for the loaded file''' + def __init__(self, name, loader, properties): + self.name = name + self.states = {'template': False, 'info': False} + self.metadata = {'owner': 'root', 'group': 'root', 'perms': '0644'} + self.properties = properties + self.loader = loader + + def HandleEvent(self, event): + '''Handle all fs events for this template''' + if event.filename in ['template.xml', 'template.txt']: + try: + if event.filename.endswith('.txt'): + self.template = self.loader.load(os.path.join(self.name[1:], event.filename), cls=TextTemplate) + else: + self.template = self.loader.load(os.path.join(self.name[1:], event.filename), cls=MarkupTemplate) + except TemplateError, terror: + logger.error('Genshi template error: %s' % terror) + + elif event.filename == 'info': + for line in open(self.name + '/info').readlines(): + match = info.match(line) + if not match: + logger.warning("Failed to match line: %s"%line) + continue + else: + mgd = match.groupdict() + if mgd['owner']: + self.metadata['owner'] = mgd['owner'] + elif mgd['group']: + self.metadata['group'] = mgd['group'] + elif mgd['perms']: + self.metadata['perms'] = mgd['perms'] + if len(self.metadata['perms']) == 3: + self.metadata['perms'] = "0%s" % (self.metadata['perms']) + else: + logger.info('Ignoring event for %s' % event.filename) + + def BuildFile(self, entry, metadata): + '''Build literal file information''' + try: + stream = self.template.generate(name=entry.get('name'), metadata=metadata, properties=self.properties) + if isinstance(self.template, TextTemplate): + entry.text = stream.render('text') + else: + entry.text = stream.render('xml') + except TemplateError, terror: + logger.error('Genshi template error: %s' % terror) + raise Bcfg2.Server.Plugin.PluginExecutionError + [entry.attrib.__setitem__(key, value) for (key, value) in self.metadata.iteritems()] + +class GenshiProperties(Bcfg2.Server.Plugin.SingleXMLFileBacked): + '''Class for Genshi properties''' + def Index(self): + '''Build data into an elementtree object for templating usage''' + try: + self.properties = lxml.etree.XML(self.data) + del self.data + except lxml.etree.XMLSyntaxError: + logger.error("Failed to parse properties") + +class FakeProperties: + '''Dummy class used when properties dont exist''' + def __init__(self): + self.properties = lxml.etree.Element("Properties") + +class TGenshi(Bcfg2.Server.Plugin.Plugin): + '''The TGenshi generator implements a templating mechanism for configuration files''' + __name__ = 'TGenshi' + __version__ = '$Id$' + __author__ = 'jeff@ocjtech.us' + + def __init__(self, core, datastore): + Bcfg2.Server.Plugin.Plugin.__init__(self, core, datastore) + if self.data[-1] == '/': + self.data = self.data[:-1] + self.Entries['ConfigFile'] = {} + self.entries = {} + self.handles = {} + self.loader = TemplateLoader(self.data, auto_reload=True) + self.AddDirectoryMonitor('') + try: + self.properties = GenshiProperties('%s/../etc/properties.xml' \ + % (self.data), self.core.fam) + except: + self.properties = FakeProperties() + self.logger.info("Failed to read properties file; TGenshi properties disabled") + + def BuildEntry(self, entry, metadata): + '''Dispatch fetch calls to the correct object''' + self.entries[entry.get('name')].BuildFile(entry, metadata) + + def HandleEvent(self, event): + '''Unified FAM event handler for DirShadow''' + action = event.code2str() + if event.filename[0] == '/': + return + epath = "".join([self.data, self.handles[event.requestID], event.filename]) + if event.filename in ['info', 'template.xml', 'template.txt']: + identifier = self.handles[event.requestID][:-1] + else: + identifier = self.handles[event.requestID] + event.filename + if action in ['exists', 'created']: + if posixpath.isdir(epath): + self.AddDirectoryMonitor(epath[len(self.data):]) + elif event.filename in ['info', 'template.xml', 'template.txt']: + if not self.entries.has_key(identifier): + self.entries[identifier] = TemplateFile(identifier, self.loader, self.properties) + self.Entries['ConfigFile'][identifier] = self.BuildEntry + self.entries[identifier].HandleEvent(event) + else: + logger.info('Not creating template for %s' % identifier) + elif action == 'changed': + if self.entries.has_key(identifier): + self.entries[identifier].HandleEvent(event) + elif action == 'deleted': + if event.filename in ['template.xml', 'template.txt'] and self.entries.has_key(identifier): + del self.entries[identifier] + del self.Entries['ConfigFile'][identifier] + + def AddDirectoryMonitor(self, relative): + '''Add new directory to FAM structures''' + if not relative: + relative = '/' + if relative[-1] != '/': + relative += '/' + name = self.data + relative + if relative not in self.handles.values(): + if not posixpath.isdir(name): + print "Genshi: Failed to open directory %s" % (name) + return + reqid = self.core.fam.AddMonitor(name, self) + self.handles[reqid] = relative diff --git a/src/lib/Server/Plugins/__init__.py b/src/lib/Server/Plugins/__init__.py index f863fdce0..484162451 100644 --- a/src/lib/Server/Plugins/__init__.py +++ b/src/lib/Server/Plugins/__init__.py @@ -2,4 +2,5 @@ __revision__ = '$Revision$' all = ['Account', 'Base', 'Bundler', 'Cfg', 'Hostbase', 'Metadata', - 'Pkgmgr', 'Rules', 'SSHbase', 'Svcmgr', 'Vhost'] + 'Pkgmgr', 'Rules', 'SSHbase', 'Svcmgr', 'TCheetah', + 'TGenshi', 'Vhost'] -- cgit v1.2.3-1-g7c22