From 521f8a1714378021a694d2b312b0281d3abf3ef7 Mon Sep 17 00:00:00 2001
From: Tim Laszlo <tim.laszlo@gmail.com>
Date: Wed, 2 Feb 2011 19:51:47 -0600
Subject: Missing newline in help

---
 src/lib/Server/Admin/Group.py      |  2 +-
 src/lib/Server/Plugins/Metadata.py | 83 +++++++++++++-------------------------
 2 files changed, 29 insertions(+), 56 deletions(-)

(limited to 'src')

diff --git a/src/lib/Server/Admin/Group.py b/src/lib/Server/Admin/Group.py
index d4024686d..4b2db28ec 100644
--- a/src/lib/Server/Admin/Group.py
+++ b/src/lib/Server/Admin/Group.py
@@ -9,7 +9,7 @@ class Group(Bcfg2.Server.Admin.MetadataCore):
                                     "\nbcfg2-admin group update <group> "
                                     "attr1=val1 attr2=val2"
                                     "\nbcfg2-admin group list"
-                                    "bcfg2-admin group del <group>\n")
+                                    "\nbcfg2-admin group del <group>\n")
     __usage__ = ("bcfg2-admin group [options] [add|del|update|list] [attr=val]")
 
     def __init__(self, configfile):
diff --git a/src/lib/Server/Plugins/Metadata.py b/src/lib/Server/Plugins/Metadata.py
index aef65f960..36ff4ee85 100644
--- a/src/lib/Server/Plugins/Metadata.py
+++ b/src/lib/Server/Plugins/Metadata.py
@@ -280,72 +280,45 @@ class Metadata(Bcfg2.Server.Plugin.Plugin,
 
     def add_group(self, group_name, attribs):
         """Add group to groups.xml."""
-        tree = lxml.etree.parse(self.data + "/groups.xml")
-        root = tree.getroot()
-        element = lxml.etree.Element("Group", name=group_name)
-        for key, val in attribs.iteritems():
-            element.set(key, val)
-        node = self.search_group(group_name, tree)
+
+        node = self.search_group(group_name, self.groups_xml.xdata)
         if node != None:
             self.logger.error("Group \"%s\" already exists" % (group_name))
             raise MetadataConsistencyError
-        root.append(element)
-        group_tree = open(self.data + "/groups.xml","w")
-        fd = group_tree.fileno()
-        while True:
-            try:
-                fcntl.lockf(fd, fcntl.LOCK_EX | fcntl.LOCK_NB)
-            except IOError:
-                continue
-            else:
-                break
-        tree.write(group_tree)
-        fcntl.lockf(fd, fcntl.LOCK_UN)
-        group_tree.close()
+
+        element = lxml.etree.SubElement(self.groups_xml.base_xdata.getroot(),
+                                      "group", name=group_name)
+        for key, val in attribs.iteritems():
+            element.set(key, val)
+        self.groups_xml.write()
 
     def update_group(self, group_name, attribs):
         """Update a groups attributes."""
-        tree = lxml.etree.parse(self.data + "/groups.xml")
-        root = tree.getroot()
-        node = self.search_group(group_name, tree)
-        if node == None:
-            self.logger.error("Group \"%s\" not found" % (group_name))
+        node = self.search_group(group_name, self.groups_xml.xdata)
+        if node != None:
+            self.logger.error("Group \"%s\" already exists" % (group_name))
             raise MetadataConsistencyError
-        node.attrib.update(attribs)
-        group_tree = open(self.data + "/groups.xml","w")
-        fd = group_tree.fileno()
-        while True:
-            try:
-                fcntl.lockf(fd, fcntl.LOCK_EX | fcntl.LOCK_NB)
-            except IOError:
-                continue
-            else:
-                break
-        tree.write(group_tree)
-        fcntl.lockf(fd, fcntl.LOCK_UN)
-        group_tree.close()
+        xdict = self.groups_xml.find_xml_for_xpath('.//Group[@name="%s"]' % (node.get('name')))
+        if not xdict:
+            self.logger.error("Unexpected error finding group")
+            raise MetadataConsistencyError
+
+        for key, val in attribs.iteritems():
+            xdict['xquery'][0].set(key, val)
+        self.groups_xml.write_xml(xdict['filename'], xdict['xmltree'])
 
     def remove_group(self, group_name):
         """Remove a group."""
-        tree = lxml.etree.parse(self.data + "/groups.xml")
-        root = tree.getroot()
-        node = self.search_group(group_name, tree)
-        if node == None:
-            self.logger.error("Client \"%s\" not found" % (group_name))
+        node = self.search_group(group_name, self.groups_xml.xdata)
+        if node != None:
+            self.logger.error("Group \"%s\" already exists" % (group_name))
             raise MetadataConsistencyError
-        root.remove(node)
-        group_tree = open(self.data + "/groups.xml","w")
-        fd = group_tree.fileno()
-        while True:
-            try:
-                fcntl.lockf(fd, fcntl.LOCK_EX | fcntl.LOCK_NB)
-            except IOError:
-                continue
-            else:
-                break
-        tree.write(group_tree)
-        fcntl.lockf(fd, fcntl.LOCK_UN)
-        group_tree.close()
+        xdict = self.groups_xml.find_xml_for_xpath('.//Group[@name="%s"]' % (node.get('name')))
+        if not xdict:
+            self.logger.error("Unexpected error finding group")
+            raise MetadataConsistencyError
+        xdict['xmltree'].remove(xdict['xquery'][0])
+        self.groups_xml.write_xml(xdict['filename'], xdict['xmltree'])
 
     def add_bundle(self, bundle_name):
         """Add bundle to groups.xml."""
-- 
cgit v1.2.3-1-g7c22