diff options
Diffstat (limited to 'src/lib/Server/Admin/Minestruct.py')
-rw-r--r-- | src/lib/Server/Admin/Minestruct.py | 28 |
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) |