fixed PropertyFile.Index() to populate .data as a side effect

1 files changed, 37 insertions, 1 deletions

diff --git a/src/lib/Server/Plugins/Properties.py b/src/lib/Server/Plugins/Properties.py
index c5d2acc44..b34bde998 100644
--- a/src/lib/Server/Plugins/Properties.py
+++ b/src/lib/Server/Plugins/Properties.py
@@ -6,7 +6,43 @@ import Bcfg2.Server.Plugin
 
 class PropertyFile(Bcfg2.Server.Plugin.StructFile):
     """Class for properties files."""
-    pass
+    def Index(self):
+        """Build internal data structures."""
+        if type(self.data) is not lxml.etree._Element:
+            try:
+                self.data = lxml.etree.XML(self.data)
+            except lxml.etree.XMLSyntaxError:
+                Bcfg2.Server.Plugin.logger.error("Failed to parse %s" %
+                                                 self.name)
+
+        self.fragments = {}
+        work = {lambda x: True: self.data.getchildren()}
+        while work:
+            (predicate, worklist) = work.popitem()
+            self.fragments[predicate] = \
+                                      [item for item in worklist
+                                       if (item.tag != 'Group' and
+                                           item.tag != 'Client' and
+                                           not isinstance(item,
+                                                          lxml.etree._Comment))]
+            for item in worklist:
+                cmd = None
+                if item.tag == 'Group':
+                    if item.get('negate', 'false').lower() == 'true':
+                        cmd = "lambda x:'%s' not in x.groups and predicate(x)"
+                    else:
+                        cmd = "lambda x:'%s' in x.groups and predicate(x)"
+                elif item.tag == 'Client':
+                    if item.get('negate', 'false').lower() == 'true':
+                        cmd = "lambda x:x.hostname != '%s' and predicate(x)"
+                    else:
+                        cmd = "lambda x:x.hostname == '%s' and predicate(x)"
+                # else, ignore item
+                if cmd is not None:
+                    newpred = eval(cmd % item.get('name'),
+                                   {'predicate':predicate})
+                    work[newpred] = item.getchildren()
+
 
 
 class PropDirectoryBacked(Bcfg2.Server.Plugin.DirectoryBacked):