Ticket #243: bcfg2-PluginTCheetah_with_Probes.patch

TCheetah.py

@@ -8 +8 @@
 info = re.compile('^owner:(\s)*(?P<owner>\w+)$|group:(\s)*(?P<group>\w+)$|' +
                   'perms:(\s)*(?P<perms>\w+)$')
 
+probeData = {}
+
 class TemplateFile:
     '''Template file creates Cheetah template structures for the loaded file'''
     def __init__(self, name, properties):
@@ -48 +50 @@
         self.template.metadata = metadata
         self.template.properties = self.properties.properties
         try:
+            self.template.probes = probeData[metadata.hostname]
+        except:
+            logger.info("No probes available in template.")
+        try:
             entry.text = str(self.template)
         except:
             logger.error("Failed to template %s" % entry.get('name'), exc_info=1)
@@ -79 +85 @@
         self.handles = {}
         self.AddDirectoryMonitor('')
         self.properties = CheetahProperties('%s/../etc/properties.xml' % self.data, self.core.fam)
+        try:
+            self.probes = Bcfg2.Server.Plugin.DirectoryBacked(datastore + '/Probes', self.core.fam )
+        except:
+            self.probes = False
+        # eventually flush fam events here so that all entries built here
+        # ready to go
 
+    def GetProbes(self, _):
+        '''Return a set of probes for execution on client'''
+        ret = []
+        bangline = re.compile('^#!(?P<interpreter>(/\w+)+)$')
+        if self.probes:
+            for name, entry in self.probes.entries.iteritems():
+                probe = lxml.etree.Element('probe')
+                probe.set('name', name )
+                probe.set('source', 'TCheetah')
+                probe.text = entry.data
+                match = bangline.match(entry.data.split('\n')[0])
+                if match:
+                    probe.set('interpreter', match.group('interpreter'))
+                ret.append(probe)
+        return ret
+
+    def ReceiveData(self, client, data):
+        '''Receive probe results pertaining to client'''
+        try:
+            probeData[client.hostname].update({ data.get('name'):data.text })
+        except KeyError:
+            probeData[client.hostname] = { data.get('name'):data.text }
+
     def BuildEntry(self, entry, metadata):
         '''Dispatch fetch calls to the correct object'''
         self.entries[entry.get('name')].BuildFile(entry, metadata)