add zabbix_host_massupdate.py

zhanghe5 · zhanghe5 · commit 498834c47bd2 · 2017-01-23T17:16:02.000+08:00
diff --git a/zabbix_host_massupdate.py b/zabbix_host_massupdate.py
@@ -0,0 +1,123 @@
+#!/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)
