[PATCH master v2 01/21] Implement GetAllDisksInfo in config

[Alex Pyrgiotis]

From: Ilias Tsitsimpis <ilias...@grnet.gr>

Implement 'GetAllDisksInfo' function which returns the configuration of
all disks.

Signed-off-by: Ilias Tsitsimpis <ilias...@grnet.gr>
Signed-off-by: Alex Pyrgiotis <apyr...@grnet.gr>

diff --git a/lib/config/__init__.py b/lib/config/__init__.py
index cca2675..16313b1 100644
--- a/lib/config/__init__.py
+++ b/lib/config/__init__.py
@@ -487,6 +487,35 @@ class ConfigWriter(object):
     """
     return self._UnlockedGetDiskInfo(disk_uuid)
 
+  def _UnlockedGetDiskList(self):
+    """Get the list of disks.
+
+    @return: array of disks, ex. ['disk2-uuid', 'disk1-uuid']
+
+    """
+    return self._ConfigData().disks.keys()
+
+  @ConfigSync(shared=1)
+  def GetAllDisksInfo(self):
+    """Get the configuration of all disks.
+
+    This is a simple wrapper over L{_UnlockedGetAllDisksInfo}.
+
+    """
+    return self._UnlockedGetAllDisksInfo()
+
+  def _UnlockedGetAllDisksInfo(self):
+    """Get the configuration of all disks.
+
+    @rtype: dict
+    @return: dict of (disk, disk_info), where disk_info is what
+        would GetDiskInfo return for the node
+
+    """
+    my_dict = dict([(disk_uuid, self._UnlockedGetDiskInfo(disk_uuid))
+                    for disk_uuid in self._UnlockedGetDiskList()])
+    return my_dict
+
   def _AllInstanceNodes(self, inst_uuid):
     """Compute the set of all disk-related nodes for an instance.
 
@@ -3491,9 +3520,6 @@ class ConfigWriter(object):
     return dict((disk_uuid, self._UnlockedGetDiskInfo(disk_uuid))
                 for disk_uuid in self._UnlockedGetDiskList())
 
-  def _UnlockedGetDiskList(self):
-    return self._ConfigData().disks.keys()
-
   @ConfigSync(shared=1)
   def GetInstanceForDisk(self, disk_uuid):
     """Returns the instance the disk is currently attached to.
-- 
1.7.10.4