1 files changed, 65 insertions, 0 deletions
diff --git a/src/lib/Bcfg2/Server/Admin/Minestruct.py b/src/lib/Bcfg2/Server/Admin/Minestruct.py
new file mode 100644
index 000000000..b929a9a8c
--- /dev/null
+++ b/src/lib/Bcfg2/Server/Admin/Minestruct.py
@@ -0,0 +1,65 @@
+import getopt
+import lxml.etree
+import sys
+
+import Bcfg2.Server.Admin
+
+class Minestruct(Bcfg2.Server.Admin.StructureMode):
+    """Pull extra entries out of statistics."""
+    __shorthelp__ = "Extract extra entry lists from statistics"
+    __longhelp__ = (__shorthelp__ +
+                    "\n\nbcfg2-admin minestruct [-f filename] "
+                    "[-g groups] client\n")
+    __usage__ = ("bcfg2-admin minestruct [options] <client>\n\n"
+                 "     %-25s%s\n"
+                 "     %-25s%s\n" %
+                ("-f <filename>",
+                 "build a particular file",
+                 "-g <groups>",
+                 "only build config for groups"))
+
+    def __call__(self, args):
+        Bcfg2.Server.Admin.Mode.__call__(self, args)
+        if len(args) == 0:
+            self.errExit("No argument specified.\n"
+                         "Please see bcfg2-admin minestruct help for usage.")
+        try:
+            (opts, args) = getopt.getopt(args, 'f:g:h')
+        except:
+            self.log.error(self.__shorthelp__)
+            raise SystemExit(1)
+
+        client = args[0]
+        output = sys.stdout
+        groups = []
+
+        for (opt, optarg) in opts:
+            if opt == '-f':
+                try:
+                    output = open(optarg, 'w')
+                except IOError:
+                    self.log.error("Failed to open file: %s" % (optarg))
+                    raise SystemExit(1)
+            elif opt == '-g':
+                groups = optarg.split(':')
+
+        try:
+            extra = set()
+            for source in self.bcore.pull_sources:
+                for item in source.GetExtra(client):
+                    extra.add(item)
+        except:
+            self.log.error("Failed to find extra entry info for client %s" %
+                            client)
+            raise SystemExit(1)
+        root = lxml.etree.Element("Base")
+        self.log.info("Found %d extra entries" % (len(extra)))
+        add_point = root
+        for g in groups:
+            add_point = lxml.etree.SubElement(add_point, "Group", name=g)
+        for tag, name in extra:
+            self.log.info("%s: %s" % (tag, name))
+            lxml.etree.SubElement(add_point, tag, name=name)
+
+        tree = lxml.etree.ElementTree(root)
+        tree.write(output, pretty_print=True)