Switch bcfg2-admin minestruct to use the new statistics api to get at data in either of statistics.xml or dbstats directly

git-svn-id: https://svn.mcs.anl.gov/repos/bcfg/trunk/bcfg2@4862 ce84e21b-d406-0410-9b95-82705330c041

1 files changed, 11 insertions, 17 deletions

diff --git a/src/lib/Server/Admin/Minestruct.py b/src/lib/Server/Admin/Minestruct.py
index 61a277d2d..b63bdd2b5 100644
--- a/src/lib/Server/Admin/Minestruct.py
+++ b/src/lib/Server/Admin/Minestruct.py
@@ -2,14 +2,11 @@
 import Bcfg2.Server.Admin
 import lxml.etree
 
-class Minestruct(Bcfg2.Server.Admin.Mode):
+class Minestruct(Bcfg2.Server.Admin.StructureMode):
     '''Pull extra entries out of statistics'''
     __shorthelp__ = 'bcfg2-admin minestruct <client> [-f file-name] [-g groups]'
     __longhelp__ = __shorthelp__ + '\n\tExtract extra entry lists from statistics'
 
-    def __init__(self, cfile):
-        Bcfg2.Server.Admin.Mode.__init__(self, cfile)
-
     def __call__(self, args):
         Bcfg2.Server.Admin.Mode.__call__(self, args)
         if len(args) == 0:
@@ -43,19 +40,13 @@ class Minestruct(Bcfg2.Server.Admin.Mode):
                 continue
             else:
                 client = arg
-        stats = self.load_stats(client)
-        if len(stats.getchildren()) == 2:
-            # client is dirty
-            current = [ent for ent in stats.getchildren() if ent.get('state') == 'dirty'][0]
-        else:
-            current = stats.getchildren()[0]
-        extra = current.find('Extra').getchildren()
+        extra = self.statistics.GetExtra(client)
         root = lxml.etree.Element("Base")
         self.log.info("Found %d extra entries" % (len(extra)))
         if len(groups) == 0:
-            for entry in extra:
-                self.log.info("%s: %s" % (entry.tag, entry.get('name')))
-                root.append(lxml.etree.Element(entry.tag, name=entry.get('name')))
+            for tag, name in extra:
+                self.log.info("%s: %s" % (tag, name))
+                lxml.etree.SubElement(root, tag, name=name)
         else:
             groups_root = lxml.etree.Element("Group", name=groups[0])
             root.append(groups_root) 
@@ -64,8 +55,9 @@ class Minestruct(Bcfg2.Server.Admin.Mode):
                 groups_root.append(temp)
                 groups_root = temp
             for entry in extra:
-                self.log.info("%s: %s" % (entry.tag, entry.get('name')))
-                groups_root.append(lxml.etree.Element(entry.tag, name=entry.get('name')))
+                self.log.info("%s: %s" % (tag, name))
+                lxml.etree.SubElement(groups_root, tag, name=name)
+
         tree = lxml.etree.ElementTree(root)
         if write_to_file == True:
             try:
@@ -73,5 +65,7 @@ class Minestruct(Bcfg2.Server.Admin.Mode):
             except IOError:
                 self.log.info("Failed to write to file: %s" % (output_file))
                 raise SystemExit(1)
-            tree.write(f)
+            tree.write(f, pretty_print=True)
+        else:
+            print lxml.etree.tostring(tree, pretty_print=True)