Add 'resource.pool.get' method for retrieving properties of a single

ResourcePool managed object
dnaeon · Mar 30, 2014 · ca13bbd · ca13bbd
1 parent 5e06694
commit ca13bbd
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 2 deletions.
diff --git a/src/vpoller/agent.py b/src/vpoller/agent.py
@@ -256,7 +256,7 @@ def cluster_get(self, msg):
         Example client message would be:
 
             {
-                "method":     "datacenter.get",
+                "method":     "cluster.get",
                 "hostname":   "vc01.example.org",
                 "name":       "MyCluster",
                 "properties": [
@@ -313,7 +313,41 @@ def resource_pool_discover(self, msg):
             properties.extend(msg['properties'])
 
         return self._discover_objects(properties=properties, obj_type=pyVmomi.vim.ResourcePool)
-
+
+    def resource_pool_get(self, msg):
+        """
+        Get properties of a single pyVmomi.vim.ResourcePool managed object
+
+        Example client message would be:
+
+            {
+                "method":     "resource.pool.get",
+                "hostname":   "vc01.example.org",
+                "name":       "MyResourcePool",
+                "properties": [
+                    "name",
+                    "runtime.cpu",
+                    "runtime.memory",
+                    "runtime.overallStatus"
+                ]
+            }
+              
+        Returns:
+            The managed object properties in JSON format
+
+        """
+        # Property names to be collected
+        properties = ['name']
+        if msg.has_key('properties') and msg['properties']:
+            properties.extend(msg['properties'])
+
+        return self._get_object_properties(
+            properties=properties,
+            obj_type=pyVmomi.vim.ResourcePool,
+            obj_property_name='name',
+            obj_property_value=msg['name']
+        )
+
     def host_discover(self, msg):
         """
         Discover all pyVmomi.vim.HostSystem managed objects

diff --git a/src/vpoller/worker.py b/src/vpoller/worker.py
@@ -310,6 +310,10 @@ def process_client_msg(self, msg):
                 'method':    self.agents[vsphere_host].resource_pool_discover,
                 'msg_attr':  ('method', 'hostname'),
             },
+            'resource.pool.get': {
+                'method':    self.agents[vsphere_host].resource_pool_get,
+                'msg_attr':  ('method', 'hostname', 'name', 'properties'),
+            },
             'host.discover': {
                 'method':    self.agents[vsphere_host].host_discover,
                 'msg_attr':  ('method', 'hostname'),