add function for servicegrup downtimes

This commit is contained in:
Michael Gröning 2014-11-07 14:14:12 +01:00 committed by Greg DeKoenigsberg
parent 1276420a3a
commit 8829b818b8

View file

@ -169,6 +169,7 @@ def main():
'silence_nagios', 'silence_nagios',
'unsilence_nagios', 'unsilence_nagios',
'command', 'command',
'servicegroup_downtime'
] ]
module = AnsibleModule( module = AnsibleModule(
@ -176,6 +177,7 @@ def main():
action=dict(required=True, default=None, choices=ACTION_CHOICES), action=dict(required=True, default=None, choices=ACTION_CHOICES),
author=dict(default='Ansible'), author=dict(default='Ansible'),
host=dict(required=False, default=None), host=dict(required=False, default=None),
servicegroup=dict(required=False, default=None),
minutes=dict(default=30), minutes=dict(default=30),
cmdfile=dict(default=which_cmdfile()), cmdfile=dict(default=which_cmdfile()),
services=dict(default=None, aliases=['service']), services=dict(default=None, aliases=['service']),
@ -185,6 +187,7 @@ def main():
action = module.params['action'] action = module.params['action']
host = module.params['host'] host = module.params['host']
servicegroup = module.params['servicegroup']
minutes = module.params['minutes'] minutes = module.params['minutes']
services = module.params['services'] services = module.params['services']
cmdfile = module.params['cmdfile'] cmdfile = module.params['cmdfile']
@ -201,7 +204,7 @@ def main():
# 'minutes' and 'service' manually. # '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: if not host:
module.fail_json(msg='no host specified for action requiring one') module.fail_json(msg='no host specified for action requiring one')
###################################################################### ######################################################################
@ -217,6 +220,20 @@ def main():
except Exception: except Exception:
module.fail_json(msg='invalid entry for minutes') 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 action in ['enable_alerts', 'disable_alerts']:
if not services: if not services:
@ -259,6 +276,7 @@ class Nagios(object):
self.action = kwargs['action'] self.action = kwargs['action']
self.author = kwargs['author'] self.author = kwargs['author']
self.host = kwargs['host'] self.host = kwargs['host']
self.service_group = kwargs['servicegroup']
self.minutes = int(kwargs['minutes']) self.minutes = int(kwargs['minutes'])
self.cmdfile = kwargs['cmdfile'] self.cmdfile = kwargs['cmdfile']
self.command = kwargs['command'] self.command = kwargs['command']
@ -847,6 +865,9 @@ class Nagios(object):
self.schedule_svc_downtime(self.host, self.schedule_svc_downtime(self.host,
services=self.services, services=self.services,
minutes=self.minutes) 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 # toggle the host AND service alerts
elif self.action == 'silence': elif self.action == 'silence':