zabbix_host_massupdate.py

#!/usr/bin/env python
# coding:utf8

'''
Created on 03.06.2015
'''

import optparse
import sys
import traceback
from getpass import getpass
from core import ZabbixAPI
import sys


def get_options():
    usage = "usage: %prog [options]"
    OptionParser = optparse.OptionParser
    parser = OptionParser(usage)

    parser.add_option("-s", "--server", action="store", type="string", \
                      dest="server", help="(REQUIRED)Zabbix Server URL.")
    parser.add_option("-u", "--username", action="store", type="string", \
                      dest="username", help="(REQUIRED)Username (Will prompt if not given).")
    parser.add_option("-p", "--password", action="store", type="string", \
                      dest="password", help="(REQUIRED)Password (Will prompt if not given).")
    parser.add_option("-H", "--hostname", action="store", type="string", \
                      dest="hostname", help="(REQUIRED)hostname for hosts.If you want to use multiple groups,separate them with a ','.")
    parser.add_option("-t", "--templates", action="store", type="string", \
                      dest="templates", default="",
                      help="Templates to replace the currently linked templates.If you want to use multiple groups,separate them with a ','.")
    parser.add_option("-g", "--groups", action="store", type="string", \
                      dest="groups", default="",
                      help="Host groups to replace the current host groups the host belongs to.If you want to use multiple groups,separate them with a ','.")
    parser.add_option("-f", "--file", dest="filename", \
                      metavar="FILE",
                      help="Load values from input file. Specify - for standard input Each line of file contains whitespace delimited: <hostname>4space<groups>4space<templates>")

    options, args = parser.parse_args()

    if not options.server:
        options.server = raw_input('server http:')

    if not options.username:
        options.username = raw_input('Username:')

    if not options.password:
        options.password = getpass()

    return options, args


def errmsg(msg):
    sys.stderr.write(msg + "\n")
    sys.exit(-1)


def get_list(lists, key):
    l = []
    for i in lists:
        l.append(i[key])
    return l


def get_api(**kwargs):
    print kwargs
    try:
        print zapi.host.massupdate(kwargs)
    except Exception as e:
        print str(e)


if __name__ == "__main__":
    options, args = get_options()

    zapi = ZabbixAPI(options.server, options.username, options.password)

    file = options.filename

    if file:
        with open(file, "r") as f:
            content = f.readlines()
            for i in content:
                l = i.split("    ")
                hostname = l[0].rstrip()
                hosts = hostname.split(',')
                hostid = zapi.host.get({"filter": {"host": hosts}})
                try:
                    templates = l[2].rstrip()
                    templates_id = zapi.template.get({"output": "templateid", "filter": {"host": templates.split(",")}})
                except:
                    templates_id = ""
                try:
                    groups = l[1].rstrip()
                    groups_id = zapi.hostgroup.get({"output": "groupid", "filter": {"name": groups.split(",")}})
                except:
                    groups_id = ""
                if templates_id and groups_id:
                    get_api(templates=templates_id, groups=groups_id, hosts=hostid)
                elif templates_id and not groups_id:
                    get_api(templates=templates_id, hosts=hostid)
                elif not templates_id and groups_id:
                    get_api(groups=groups_id, hosts=hostid)
    else:
        templates = options.templates
        hostname = options.hostname
        groups = options.groups
        hosts = hostname.split(',')
        hostid = zapi.host.get({"filter": {"host": hosts}})
        if templates:
            templates_id = zapi.template.get({"output": "templateid", "filter": {"host": templates.split(",")}})
        else:
            templates_id = ""
        if groups:
            groups_id = zapi.hostgroup.get({"output": "groupid", "filter": {"name": groups.split(",")}})
        else:
            groups_id = ""
        if templates_id and groups_id:
            get_api(templates=templates_id, groups=groups_id, hosts=hostid)
        elif templates_id and not groups_id:
            get_api(templates=templates_id, hosts=hostid)
        elif not templates_id and groups_id:
            get_api(groups=groups_id, hosts=hostid)