From 8a209ed30b0ecf4b5c4f5cc208e9ee373de426b7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Michael=20Gr=C3=B6ning?= <mgroe@localhost.localdomain>
Date: Fri, 7 Nov 2014 14:14:12 +0100
Subject: [PATCH] add function for servicegrup downtimes

---
 .../modules/extras/monitoring/nagios.py       | 23 ++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/lib/ansible/modules/extras/monitoring/nagios.py b/lib/ansible/modules/extras/monitoring/nagios.py
index bfa498496e6..37d1ad29a91 100644
--- a/lib/ansible/modules/extras/monitoring/nagios.py
+++ b/lib/ansible/modules/extras/monitoring/nagios.py
@@ -178,6 +178,7 @@ def main():
         'silence_nagios',
         'unsilence_nagios',
         'command',
+        'servicegroup_downtime'
         ]
 
     module = AnsibleModule(
@@ -186,6 +187,7 @@ def main():
             author=dict(default='Ansible'),
             comment=dict(default='Scheduling downtime'),
             host=dict(required=False, default=None),
+            servicegroup=dict(required=False, default=None),
             minutes=dict(default=30),
             cmdfile=dict(default=which_cmdfile()),
             services=dict(default=None, aliases=['service']),
@@ -195,6 +197,7 @@ def main():
 
     action = module.params['action']
     host = module.params['host']
+    servicegroup = module.params['servicegroup']
     minutes = module.params['minutes']
     services = module.params['services']
     cmdfile = module.params['cmdfile']
@@ -211,7 +214,7 @@ def main():
     # 'minutes' and 'service' manually.
 
     ##################################################################
-    if action not in ['command', 'silence_nagios', 'unsilence_nagios']:
+    if action not in ['command', 'silence_nagios', 'unsilence_nagios', 'servicegroup_downtime']:
         if not host:
             module.fail_json(msg='no host specified for action requiring one')
     ######################################################################
@@ -227,6 +230,20 @@ def main():
         except Exception:
             module.fail_json(msg='invalid entry for minutes')
 
+    ######################################################################
+
+    if action == 'servicegroup_downtime':
+        # Make sure there's an actual service selected
+        if not servicegroup:
+            module.fail_json(msg='no servicegroup selected to set downtime for')
+        # Make sure minutes is a number
+        try:
+            m = int(minutes)
+            if not isinstance(m, types.IntType):
+                module.fail_json(msg='minutes must be a number')
+        except Exception:
+            module.fail_json(msg='invalid entry for minutes')
+
     ##################################################################
     if action in ['enable_alerts', 'disable_alerts']:
         if not services:
@@ -270,6 +287,7 @@ class Nagios(object):
         self.author = kwargs['author']
         self.comment = kwargs['comment']
         self.host = kwargs['host']
+        self.service_group = kwargs['servicegroup']
         self.minutes = int(kwargs['minutes'])
         self.cmdfile = kwargs['cmdfile']
         self.command = kwargs['command']
@@ -861,6 +879,9 @@ class Nagios(object):
                 self.schedule_svc_downtime(self.host,
                                            services=self.services,
                                            minutes=self.minutes)
+        if self.action == "servicegroup_downtime":
+            if self.services  == 'servicegroup':
+                self.schedule_servicegroup_host_downtime(self, self.servicegroup, minutes=30)
 
         # toggle the host AND service alerts
         elif self.action == 'silence':