Allow to run directly from a git checkout (#1037)

Signed-off-by: Sol Jerome <sol.jerome@gmail.com>

1 files changed, 53 insertions, 0 deletions

diff --git a/src/lib/Bcfg2/Server/Plugins/Defaults.py b/src/lib/Bcfg2/Server/Plugins/Defaults.py
new file mode 100644
index 000000000..718192e2a
--- /dev/null
+++ b/src/lib/Bcfg2/Server/Plugins/Defaults.py
@@ -0,0 +1,53 @@
+"""This generator provides rule-based entry mappings."""
+
+import re
+import Bcfg2.Server.Plugin
+import Bcfg2.Server.Plugins.Rules
+
+class Defaults(Bcfg2.Server.Plugins.Rules.Rules,
+               Bcfg2.Server.Plugin.StructureValidator):
+    """Set default attributes on bound entries"""
+    name = 'Defaults'
+    __author__ = 'bcfg-dev@mcs.anl.gov'
+
+    # Rules is a Generator that happens to implement all of the
+    # functionality we want, so we overload it, but Defaults should
+    # _not_ handle any entries; it does its stuff in the structure
+    # validation phase.  so we overload Handle(s)Entry and HandleEvent
+    # to ensure that Defaults handles no entries, even though it's a
+    # Generator.
+
+    def HandlesEntry(self, entry, metadata):
+        return False
+
+    def HandleEntry(self, entry, metadata):
+        raise PluginExecutionError
+
+    def HandleEvent(self, event):
+        Bcfg2.Server.Plugin.XMLDirectoryBacked.HandleEvent(self, event)
+
+    def validate_structures(self, metadata, structures):
+        """ Apply defaults """
+        for struct in structures:
+            for entry in struct.iter():
+                if entry.tag.startswith("Bound"):
+                    is_bound = True
+                    entry.tag = entry.tag[5:]
+                else:
+                    is_bound = False
+                try:
+                    try:
+                        self.BindEntry(entry, metadata)
+                    except Bcfg2.Server.Plugin.PluginExecutionError:
+                        # either no matching defaults (which is okay),
+                        # or multiple matching defaults (which is not
+                        # okay, but is logged).  either way, we don't
+                        # care about the error.
+                        pass
+                finally:
+                    if is_bound:
+                        entry.tag = "Bound" + entry.tag
+
+    def _regex_enabled(self):
+        """ Defaults depends on regex matching, so force it enabled """
+        return True