add function for servicegrup downtimes

This commit is contained in:
Michael Gröning 2014-11-07 14:14:12 +01:00 committed by Matt Clay
parent edf63b1e60
commit 8a209ed30b

View file

@ -178,6 +178,7 @@ def main():
'silence_nagios', 'silence_nagios',
'unsilence_nagios', 'unsilence_nagios',
'command', 'command',
'servicegroup_downtime'
] ]
module = AnsibleModule( module = AnsibleModule(
@ -186,6 +187,7 @@ def main():
author=dict(default='Ansible'), author=dict(default='Ansible'),
comment=dict(default='Scheduling downtime'), comment=dict(default='Scheduling downtime'),
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']),
@ -195,6 +197,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']
@ -211,7 +214,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')
###################################################################### ######################################################################
@ -227,6 +230,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:
@ -270,6 +287,7 @@ class Nagios(object):
self.author = kwargs['author'] self.author = kwargs['author']
self.comment = kwargs['comment'] self.comment = kwargs['comment']
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']
@ -861,6 +879,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':