zabbix_maintenance small stylistic updates and ansible version bump

reviewable/pr18780/r1
abulimov 10 years ago committed by Michael DeHaan
parent a8d5a3f849
commit 6834ee7977

@ -25,7 +25,7 @@ module: zabbix_maintenance
short_description: Create Zabbix maintenance windows short_description: Create Zabbix maintenance windows
description: description:
- This module will let you create Zabbix maintenance windows. - This module will let you create Zabbix maintenance windows.
version_added: "1.6" version_added: "1.7"
author: Alexander Bulimov author: Alexander Bulimov
requirements: requirements:
- zabbix-api python module - zabbix-api python module
@ -55,17 +55,21 @@ options:
default: null default: null
host_names: host_names:
description: description:
- Hosts to manage maintenance window for. Separate multiple hosts with commas. - Hosts to manage maintenance window for.
Separate multiple hosts with commas.
C(host_name) is an alias for C(host_names). C(host_name) is an alias for C(host_names).
B(Required) option when C(state) is I(present) and no C(host_groups) specified. B(Required) option when C(state) is I(present)
and no C(host_groups) specified.
required: false required: false
default: null default: null
aliases: [ "host_name" ] aliases: [ "host_name" ]
host_groups: host_groups:
description: description:
- Host groups to manage maintenance window for. Separate multiple groups with commas. - Host groups to manage maintenance window for.
Separate multiple groups with commas.
C(host_group) is an alias for C(host_groups). C(host_group) is an alias for C(host_groups).
B(Required) option when C(state) is I(present) and no C(host_names) specified. B(Required) option when C(state) is I(present)
and no C(host_names) specified.
required: false required: false
default: null default: null
aliases: [ "host_group" ] aliases: [ "host_group" ]
@ -138,7 +142,8 @@ EXAMPLES = '''
login_password=pAsSwOrD login_password=pAsSwOrD
''' '''
import datetime, time import datetime
import time
try: try:
from zabbix_api import ZabbixAPI from zabbix_api import ZabbixAPI
@ -146,6 +151,7 @@ try:
except ImportError: except ImportError:
HAS_ZABBIX_API = False HAS_ZABBIX_API = False
def create_maintenance(zbx, group_ids, host_ids, start_time, maintenance_type, period, name, desc): def create_maintenance(zbx, group_ids, host_ids, start_time, maintenance_type, period, name, desc):
end_time = start_time + period end_time = start_time + period
try: try:
@ -169,6 +175,7 @@ def create_maintenance(zbx, group_ids, host_ids, start_time, maintenance_type, p
return 1, None, str(e) return 1, None, str(e)
return 0, None, None return 0, None, None
def get_maintenance_id(zbx, name): def get_maintenance_id(zbx, name):
try: try:
result = zbx.maintenance.get( result = zbx.maintenance.get(
@ -188,13 +195,15 @@ def get_maintenance_id(zbx, name):
return 0, maintenance_ids, None return 0, maintenance_ids, None
def delete_maintenance(zbx, maintenance_id): def delete_maintenance(zbx, maintenance_id):
try: try:
result = zbx.maintenance.delete(maintenance_id) zbx.maintenance.delete(maintenance_id)
except BaseException as e: except BaseException as e:
return 1, None, str(e) return 1, None, str(e)
return 0, None, None return 0, None, None
def check_maintenance(zbx, name): def check_maintenance(zbx, name):
try: try:
result = zbx.maintenance.exists( result = zbx.maintenance.exists(
@ -206,6 +215,7 @@ def check_maintenance(zbx, name):
return 1, None, str(e) return 1, None, str(e)
return 0, result, None return 0, result, None
def get_group_ids(zbx, host_groups): def get_group_ids(zbx, host_groups):
group_ids = [] group_ids = []
for group in host_groups: for group in host_groups:
@ -223,12 +233,13 @@ def get_group_ids(zbx, host_groups):
return 1, None, str(e) return 1, None, str(e)
if not result: if not result:
return 1, None, "Group id for group %s not found"%group return 1, None, "Group id for group %s not found" % group
group_ids.append(result[0]["groupid"]) group_ids.append(result[0]["groupid"])
return 0, group_ids, None return 0, group_ids, None
def get_host_ids(zbx, host_names): def get_host_ids(zbx, host_names):
host_ids = [] host_ids = []
for host in host_names: for host in host_names:
@ -246,7 +257,7 @@ def get_host_ids(zbx, host_names):
return 1, None, str(e) return 1, None, str(e)
if not result: if not result:
return 1, None, "Host id for host %s not found"%host return 1, None, "Host id for host %s not found" % host
host_ids.append(result[0]["hostid"]) host_ids.append(result[0]["hostid"])
@ -288,12 +299,11 @@ def main():
else: else:
maintenance_type = 1 maintenance_type = 1
try: try:
zbx = ZabbixAPI(server_url) zbx = ZabbixAPI(server_url)
zbx.login(login_user, login_password) zbx.login(login_user, login_password)
except BaseException as e: except BaseException as e:
module.fail_json(msg="Failed to connect to Zabbix server: %s"%e) module.fail_json(msg="Failed to connect to Zabbix server: %s" % e)
changed = False changed = False
@ -301,25 +311,25 @@ def main():
now = datetime.datetime.now() now = datetime.datetime.now()
start_time = time.mktime(now.timetuple()) start_time = time.mktime(now.timetuple())
period = 60 * int(minutes) #N * 60 seconds period = 60 * int(minutes) # N * 60 seconds
if host_groups: if host_groups:
(rc, group_ids, error) = get_group_ids(zbx, host_groups) (rc, group_ids, error) = get_group_ids(zbx, host_groups)
if rc != 0: if rc != 0:
module.fail_json(msg="Failed to get group_ids: %s"%error) module.fail_json(msg="Failed to get group_ids: %s" % error)
else: else:
group_ids = [] group_ids = []
if host_names: if host_names:
(rc, host_ids, error) = get_host_ids(zbx, host_names) (rc, host_ids, error) = get_host_ids(zbx, host_names)
if rc != 0: if rc != 0:
module.fail_json(msg="Failed to get host_ids: %s"%error) module.fail_json(msg="Failed to get host_ids: %s" % error)
else: else:
host_ids = [] host_ids = []
(rc, exists, error) = check_maintenance(zbx, name) (rc, exists, error) = check_maintenance(zbx, name)
if rc != 0: if rc != 0:
module.fail_json(msg="Failed to check maintenance %s existance: %s"%(name, error)) module.fail_json(msg="Failed to check maintenance %s existance: %s" % (name, error))
if not exists: if not exists:
if not host_names and not host_groups: if not host_names and not host_groups:
@ -332,18 +342,18 @@ def main():
if rc == 0: if rc == 0:
changed = True changed = True
else: else:
module.fail_json(msg="Failed to create maintenance: %s"%error) module.fail_json(msg="Failed to create maintenance: %s" % error)
if state == "absent": if state == "absent":
(rc, exists, error) = check_maintenance(zbx, name) (rc, exists, error) = check_maintenance(zbx, name)
if rc != 0: if rc != 0:
module.fail_json(msg="Failed to check maintenance %s existance: %s"%(name, error)) module.fail_json(msg="Failed to check maintenance %s existance: %s" % (name, error))
if exists: if exists:
(rc, maintenance, error) = get_maintenance_id(zbx, name) (rc, maintenance, error) = get_maintenance_id(zbx, name)
if rc != 0: if rc != 0:
module.fail_json(msg="Failed to get maintenance id: %s"%error) module.fail_json(msg="Failed to get maintenance id: %s" % error)
if maintenance: if maintenance:
if module.check_mode: if module.check_mode:
@ -353,8 +363,7 @@ def main():
if rc == 0: if rc == 0:
changed = True changed = True
else: else:
module.fail_json(msg="Failed to remove maintenance: %s"%error) module.fail_json(msg="Failed to remove maintenance: %s" % error)
module.exit_json(changed=changed) module.exit_json(changed=changed)

Loading…
Cancel
Save