Implement a 'datacenter.alarm.get' vPoller method

dnaeon · Sep 26, 2014 · 3583968 · 3583968
1 parent 59ce881
commit 3583968
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/docs/methods.rst b/docs/methods.rst
@@ -28,6 +28,8 @@ with description for each of them.
 +------------------------+--------------------------------------------------------------------------+
 | datacenter.get         | Get properties of a vim.Datacenter managed object                        |
 +------------------------+--------------------------------------------------------------------------+
+| datacenter.alarm.get   | Get all alarms for a vim.Datacenter managed object                       |
++------------------------+--------------------------------------------------------------------------+
 | cluster.discover       | Discover all vim.ClusterComputeResource managed objects                  |
 +------------------------+--------------------------------------------------------------------------+
 | cluster.get            | Get properties of a vim.ClusterComputeResource managed object            |

diff --git a/src/vpoller/agent.py b/src/vpoller/agent.py
@@ -115,6 +115,10 @@ def __init__(self, user, pwd, host):
                 'method': self.datacenter_get,
                 'required': ['hostname', 'name', 'properties'],
             },
+            'datacenter.alarm.get': {
+                'method': self.datacenter_alarm_get,
+                'required': ['hostname', 'name'],
+            },
             'cluster.discover': {
                 'method': self.cluster_discover,
                 'required': ['hostname'],
@@ -897,6 +901,30 @@ def datacenter_get(self, msg):
             obj_property_value=msg['name']
         )
 
+    def datacenter_alarm_get(self, msg):
+        """
+        Get all alarms for a vim.Datacenter managed object
+
+        Example client message would be:
+
+            {
+                "method":   "datacenter.alarm.get",
+                "hostname": "vc01.example.org",
+                "name":     "MyDatacenter"
+            }
+
+        Returns:
+            The discovered alarms in JSON format
+
+        """
+        result = self._object_alarm_get(
+            obj_type=pyVmomi.vim.Datacenter,
+            obj_property_name='name',
+            obj_property_value=msg['name']
+        )
+
+        return result
+
     def cluster_discover(self, msg):
         """
         Discover all vim.ClusterComputeResource managed objects