mirror of https://github.com/ansible/ansible.git
add zabbix_hostmacro
parent
400166a655
commit
d0256c593a
@ -0,0 +1,233 @@
|
||||
#!/usr/bin/python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
# (c) 2013-2014, Epic Games, Inc.
|
||||
#
|
||||
# This file is part of Ansible
|
||||
#
|
||||
# Ansible is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# Ansible is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
|
||||
DOCUMENTATION = '''
|
||||
---
|
||||
module: zabbix_hostmacro
|
||||
short_description: Zabbix host macro creates/updates/deletes
|
||||
description:
|
||||
- When the host macro does not exists, a new macro will be created, added to specific host.
|
||||
- When the host macro already exists, the value will be updated.
|
||||
- Delete a host macro from Zabbix if the macro exists.
|
||||
version_added: "1.9"
|
||||
author: Dean Hailin Song
|
||||
requirements:
|
||||
- zabbix-api python module
|
||||
options:
|
||||
server_url:
|
||||
description:
|
||||
- Url of Zabbix server, with protocol (http or https).
|
||||
C(url) is an alias for C(server_url).
|
||||
required: true
|
||||
default: null
|
||||
aliases: [ "url" ]
|
||||
login_user:
|
||||
description:
|
||||
- Zabbix user name.
|
||||
required: true
|
||||
default: null
|
||||
login_password:
|
||||
description:
|
||||
- Zabbix user password.
|
||||
required: true
|
||||
default: null
|
||||
host_name:
|
||||
description:
|
||||
- Technical name of the host.
|
||||
- If the host has already been added, the host name won't be updated.
|
||||
required: true
|
||||
macro_name:
|
||||
description:
|
||||
- Technical name of the host macro.
|
||||
required: true
|
||||
macro_value:
|
||||
description:
|
||||
- Value of the host macro.
|
||||
required: true
|
||||
state:
|
||||
description:
|
||||
- create/update or delete macro.
|
||||
- Possible values are: present and absent. If the macro already exists, and the state is "present", just to update the macro.
|
||||
required: false
|
||||
default: "present"
|
||||
timeout:
|
||||
description:
|
||||
- The timeout of API request(seconds).
|
||||
default: 10
|
||||
'''
|
||||
|
||||
EXAMPLES = '''
|
||||
- name: Create a new host macro or update an existing macro's value
|
||||
local_action:
|
||||
module: zabbix_hostmacro
|
||||
server_url: http://monitor.example.com
|
||||
login_user: username
|
||||
login_password: password
|
||||
host_name: ExampleHost
|
||||
macro_name:Example macro
|
||||
macro_value:Example value
|
||||
state: present
|
||||
'''
|
||||
|
||||
import logging
|
||||
import copy
|
||||
from ansible.module_utils.basic import *
|
||||
|
||||
try:
|
||||
from zabbix_api import ZabbixAPI, ZabbixAPISubClass
|
||||
|
||||
HAS_ZABBIX_API = True
|
||||
except ImportError:
|
||||
HAS_ZABBIX_API = False
|
||||
|
||||
|
||||
# Extend the ZabbixAPI
|
||||
# Since the zabbix-api python module too old (version 1.0, no higher version so far).
|
||||
class ZabbixAPIExtends(ZabbixAPI):
|
||||
def __init__(self, server, timeout, **kwargs):
|
||||
ZabbixAPI.__init__(self, server, timeout=timeout)
|
||||
|
||||
|
||||
class HostMacro(object):
|
||||
def __init__(self, module, zbx):
|
||||
self._module = module
|
||||
self._zapi = zbx
|
||||
|
||||
# exist host
|
||||
def is_host_exist(self, host_name):
|
||||
result = self._zapi.host.exists({'host': host_name})
|
||||
return result
|
||||
|
||||
# get host id by host name
|
||||
def get_host_id(self, host_name):
|
||||
try:
|
||||
host_list = self._zapi.host.get({'output': 'extend', 'filter': {'host': host_name}})
|
||||
if len(host_list) < 1:
|
||||
self._module.fail_json(msg="Host not found: %s" % host_name)
|
||||
else:
|
||||
host_id = host_list[0]['hostid']
|
||||
return host_id
|
||||
except Exception, e:
|
||||
self._module.fail_json(msg="Failed to get the host %s id: %s." % (host_name, e))
|
||||
|
||||
# get host macro
|
||||
def get_host_macro(self, macro_name, host_id):
|
||||
try:
|
||||
host_macro_list = self._zapi.usermacro.get(
|
||||
{"output": "extend", "selectSteps": "extend", 'hostids': [host_id], 'filter': {'macro': '{$' + macro_name + '}'}})
|
||||
if len(host_macro_list) > 0:
|
||||
return host_macro_list[0]
|
||||
return None
|
||||
except Exception, e:
|
||||
self._module.fail_json(msg="Failed to get host macro %s: %s" % (macro_name, e))
|
||||
|
||||
# create host macro
|
||||
def create_host_macro(self, macro_name, macro_value, host_id):
|
||||
try:
|
||||
if self._module.check_mode:
|
||||
self._module.exit_json(changed=True)
|
||||
self._zapi.usermacro.create({'hostid': host_id, 'macro': '{$' + macro_name + '}', 'value': macro_value})
|
||||
self._module.exit_json(changed=True, result="Successfully added host macro %s " % macro_name)
|
||||
except Exception, e:
|
||||
self._module.fail_json(msg="Failed to create host macro %s: %s" % (macro_name, e))
|
||||
|
||||
# update host macro
|
||||
def update_host_macro(self, host_macro_obj, macro_name, macro_value):
|
||||
host_macro_id = host_macro_obj['hostmacroid']
|
||||
try:
|
||||
if self._module.check_mode:
|
||||
self._module.exit_json(changed=True)
|
||||
self._zapi.usermacro.update({'hostmacroid': host_macro_id, 'value': macro_value})
|
||||
self._module.exit_json(changed=True, result="Successfully updated host macro %s " % macro_name)
|
||||
except Exception, e:
|
||||
self._module.fail_json(msg="Failed to updated host macro %s: %s" % (macro_name, e))
|
||||
|
||||
# delete host macro
|
||||
def delete_host_macro(self, host_macro_obj, macro_name):
|
||||
host_macro_id = host_macro_obj['hostmacroid']
|
||||
try:
|
||||
if self._module.check_mode:
|
||||
self._module.exit_json(changed=True)
|
||||
self._zapi.usermacro.delete([host_macro_id])
|
||||
self._module.exit_json(changed=True, result="Successfully deleted host macro %s " % macro_name)
|
||||
except Exception, e:
|
||||
self._module.fail_json(msg="Failed to delete host macro %s: %s" % (macro_name, e))
|
||||
|
||||
def main():
|
||||
module = AnsibleModule(
|
||||
argument_spec=dict(
|
||||
server_url=dict(required=True, default=None, aliases=['url']),
|
||||
login_user=dict(required=True),
|
||||
login_password=dict(required=True),
|
||||
host_name=dict(required=True),
|
||||
macro_name=dict(required=True),
|
||||
macro_value=dict(required=True),
|
||||
state=dict(default="present"),
|
||||
timeout=dict(default=10)
|
||||
),
|
||||
supports_check_mode=True
|
||||
)
|
||||
|
||||
if not HAS_ZABBIX_API:
|
||||
module.fail_json(msg="Missing requried zabbix-api module (check docs or install with: pip install zabbix-api)")
|
||||
|
||||
server_url = module.params['server_url']
|
||||
login_user = module.params['login_user']
|
||||
login_password = module.params['login_password']
|
||||
host_name = module.params['host_name']
|
||||
macro_name = (module.params['macro_name']).upper()
|
||||
macro_value = module.params['macro_value']
|
||||
state = module.params['state']
|
||||
timeout = module.params['timeout']
|
||||
|
||||
zbx = None
|
||||
# login to zabbix
|
||||
try:
|
||||
zbx = ZabbixAPIExtends(server_url, timeout=timeout)
|
||||
zbx.login(login_user, login_password)
|
||||
except Exception, e:
|
||||
module.fail_json(msg="Failed to connect to Zabbix server: %s" % e)
|
||||
|
||||
host_macro_class_obj = HostMacro(module, zbx)
|
||||
|
||||
changed = False
|
||||
|
||||
if host_name:
|
||||
host_id = host_macro_class_obj.get_host_id(host_name)
|
||||
host_macro_obj = host_macro_class_obj.get_host_macro(macro_name, host_id)
|
||||
|
||||
if state == 'absent':
|
||||
if not host_macro_obj:
|
||||
module.exit_json(changed=False, msg="Host Macro %s does not exist" % macro_name)
|
||||
else:
|
||||
# delete a macro
|
||||
host_macro_class_obj.delete_host_macro(host_macro_obj, macro_name)
|
||||
else:
|
||||
if not host_macro_obj:
|
||||
# create host macro
|
||||
host_macro_class_obj.create_host_macro(macro_name, macro_value, host_id)
|
||||
else:
|
||||
# update host macro
|
||||
host_macro_class_obj.update_host_macro(host_macro_obj, macro_name, macro_value)
|
||||
|
||||
from ansible.module_utils.basic import *
|
||||
main()
|
||||
|
Loading…
Reference in New Issue