Add 'vm.discover' method for discovering VirtualMachine managed objects

dnaeon · Mar 29, 2014 · fdd5bff · fdd5bff
1 parent bca0e46
commit fdd5bff
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/src/vpoller/agent.py b/src/vpoller/agent.py
@@ -220,3 +220,35 @@ def host_discover(self, msg):
 
         return self._discover_objects(properties=properties, obj_type=pyVmomi.vim.HostSystem)
 
+    def vm_discover(self, msg):
+        """
+        Discover all pyVmomi.vim.VirtualMachine managed objects
+
+        Example client message would be:
+        
+            {
+                "method":   "vm.discover",
+        	"hostname": "vc01.example.org",
+            }
+
+        Example client message which also requests additional properties:
+
+            {
+                "method":     "vm.discover",
+                "hostname":   "vc01.example.org",
+                "properties": [
+                    "name",
+                    "runtime.powerState"
+                ]
+            }
+              
+        Returns:
+            The discovered objects in JSON format
+
+        """
+        # Property names to be collected
+        properties = ['name']
+        if msg.has_key('properties') and msg['properties']:
+            properties.extend(msg['properties'])
+
+        return self._discover_objects(properties=properties, obj_type=pyVmomi.vim.VirtualMachine)
diff --git a/src/vpoller/worker.py b/src/vpoller/worker.py
@@ -287,6 +287,10 @@ def process_client_msg(self, msg):
                 'method':    self.agents[vsphere_host].host_discover,
                 'msg_attr':  ('method', 'hostname'),
             },
+            'vm.discover': {
+                'method':    self.agents[vsphere_host].vm_discover,
+                'msg_attr':  ('method', 'hostname'),
+            },
         }
 
         if msg['method'] not in methods: