使用python脚本自动化添加zabbix

baiyunjn

　　在zabbix上不能汇总监控图是个很苦逼的事情，有时候会妨碍到运维的工作效率，而且不能做到对比效果，但是把监控图片汇总到screen上就好看多了，而且也减小了运维每天巡检主机的时间

　　下面不多说直接上脚本：
(env1) ➜  ~ cat zabbix_add_screen.py
#!/usr/bin/python
#-*- coding:utf-8 -*-
#__author__ == 'chenmingle'
import urllib2
import sys
import json
import argparse
#定义通过HTTP方式访问API地址的函数，后面每次请求API的各个方法都会调用这个函数
def requestJson(url,values):
    data = json.dumps(values)
    req = urllib2.Request(url, data, {'Content-Type': 'application/json-rpc'})
    response = urllib2.urlopen(req, data)
    output = json.loads(response.read())
#    print output
    try:
        message = output['result']
    except:
        message = output['error']['data']
        print message
        quit()
    return output['result']
#API接口认证的函数，登录成功会返回一个Token
def authenticate(url, username, password):
    values = {'jsonrpc': '2.0',
              'method': 'user.login',
              'params': {
                  'user': username,
                  'password': password
              },
              'id': '0'
              }
    idvalue = requestJson(url,values)
    return idvalue
#定义更加主机分组名称获取各个hostid的函数
def getHosts(groupname,url,auth):
    host_list = []
    values = {'jsonrpc': '2.0',
              'method': 'hostgroup.get',
              'params': {
                  'output': 'extend',
                  'filter': {
                      'name': groupname
                  },
                  'selectHosts' : ['hostid','host'],
              },
              'auth': auth,
              'id': '2'
              }
    output = requestJson(url,values)
    for host in output[0]['hosts']:
        host_list.append(host['hostid'])
    return host_list
#定义获取graphid的函数
def getGraphs(host_list,name_list, url, auth, columns, graphtype=0 ,dynamic=0):
    if (graphtype == 0):
       selecttype = ['graphid']
       select = 'selectGraphs'
    if (graphtype == 1):
       selecttype = ['itemid', 'value_type']
       select = 'selectItems'
    values=({'jsonrpc' : '2.0',
             'method' : 'graph.get',
             'params' : {
                  'output' : ['graphid','name'],
                  select : [selecttype,'name'],
                  'hostids' : host_list,
                  'sortfield' : 'name',
                  'filter' : {
                         'name' : name_list,
                             },
                        },
             'auth' : auth,
             'id' : 3
              })
    output = requestJson(url,values)
    bb = sorted(output,key = lambda x:x['graphid'])
    graphs = []
    if (graphtype == 0):
        for i in bb:
            print i
            graphs.append(i['graphid'])
    if (graphtype == 1):
        for i in bb:
            if int(i['value_type']) in (0, 3):
               graphs.append(i['itemid'])
    graph_list = []
    x = 0
    y = 0
    for graph in graphs:
        print "x is " + str(x)
        print "y is " + str(y)
        graph_list.append({
            "resourcetype": graphtype,
            "resourceid": graph,
            "width": "500",
            "height": "200",
            "x": str(x),
            "y": str(y),
            "colspan": "0",
            "rowspan": "0",
            "elements": "0",
            "valign": "0",
            "halign": "0",
            "style": "0",
            "url": "",
            "dynamic": str(dynamic)
        })
        x += 1
#        print type(x)
#        print type(columns)
        if x == int(columns):
            x = 0
            y += 1
#    print graph_list
    return graph_list
#定义创建screen的函数
def screenCreate(url, auth, screen_name, graphids, columns):
    columns = int(columns)
    if len(graphids) % columns == 0:
        vsize = len(graphids) / columns
    else:
        vsize = (len(graphids) / columns) + 1
#先使用screen.get判断给定的screen name是否存在
    values0 = {
               "jsonrpc" : "2.0",
               "method"  : "screen.get",
               "params"  : {
                   "output" : "extend",
                   "filter" : {
                       "name" : screen_name,
                              }
                           },
               "auth" : auth,
               "id" : 2
               }
    values = {
              "jsonrpc": "2.0",
              "method": "screen.create",
              "params": {
                  "name": screen_name,
                  "hsize": columns,
                  "vsize": vsize,
                  "screenitems": []
              },
              "auth": auth,
              "id": 2
              }
    output0 = requestJson(url,values0)
    print output0
#如果给定的screen name不存在则直接创建screen
    if output0 == []:
       print "The Given Screen Name Not Exists"
       print "Creating Screen %s" %screen_name
       for i in graphids:
          values['params']['screenitems'].append(i)
       output = requestJson(url,values)
    else:
#如果给定的screen name已经存在，直接创建screen是不行的，
#要么先使用screen.delete把原来的screen删除掉，然后再创建，
#要么直接使用screen.update更新原来那个screen，
#使用screen.delete会产生新的screenid,
#使用screen.update比较合理一点。
       print "The Given Screen Name Already Exists"
       update_screenid=output0[0]["screenid"]
       print update_screenid
       print "Updating Screen Name %s  Screen ID %s" %(screen_name,update_screenid)
       values1 = {
               "jsonrpc" : "2.0",
               "method"  : "screen.update",
               "params"  : {
                       "screenid" : update_screenid,
                       "screenitems": []
                           },
               "auth"    : auth,
               "id"      : 2
                 }
       output1 = requestJson(url,values1)
       print output1
       print "Updating  Screen Name %s" %screen_name
       for i in graphids:
          values1['params']['screenitems'].append(i)
       output = requestJson(url,values1)
def main():
    url = 'http://www.***.com:80/zabbix/api_jsonrpc.php'
    username = 'Admin'
    password = 'Password'
    auth = authenticate(url, username, password)
    host_list = getHosts(groupname,url,auth)
#    print host_list
    graph_ids = getGraphs(host_list,graphname, url, auth, columns)
    screenCreate(url, auth, screenname, graph_ids, columns)
if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Create Zabbix screen from all of a host Items or Graphs.')
    parser.add_argument('-G', dest='graphname', nargs='+',metavar=('grah name'),
                        help='Zabbix Host Graph to create screen from')
#    parser.add_argument('-H', dest='hostname', nargs='+',metavar=('10.19.111.145'),
#                        help='Zabbix Host to create screen from')
    parser.add_argument('-g', dest='groupname', nargs='+',metavar=('linux server'),
                        help='Zabbix Group to create screen from')
    parser.add_argument('-n', dest='screenname', type=str,
                        help='Screen name in Zabbix.  Put quotes around it if you want spaces in the name.')
    parser.add_argument('-c', dest='columns', type=int,
                        help='number of columns in the screen')
    args = parser.parse_args()
    print args
#    hostname = args.hostname
    groupname = args.groupname
    screenname = args.screenname
    columns = args.columns
    graphname = args.graphname
    if columns is None:
        columns = len(graphname)
#    print columns
    main()　　最后执行下脚本
(env1) ➜  ~ python zabbix_add_screen.py -g testENV -G 'Network traffic on eth0' -n 'test Netowrk eth0' -c 2
Namespace(columns=2, graphname=['Network traffic on eth0'], groupname=['testENV'], screenname='test Netowrk eth0')
{u'graphid': u'1297', u'name': u'Network traffic on eth0'}
{u'graphid': u'4030', u'name': u'Network traffic on eth0'}
{u'graphid': u'4396', u'name': u'Network traffic on eth0'}
{u'graphid': u'4584', u'name': u'Network traffic on eth0'}
{u'graphid': u'7567', u'name': u'Network traffic on eth0'}
{u'graphid': u'7602', u'name': u'Network traffic on eth0'}
{u'graphid': u'7639', u'name': u'Network traffic on eth0'}
{u'graphid': u'7645', u'name': u'Network traffic on eth0'}
{u'graphid': u'7648', u'name': u'Network traffic on eth0'}
{u'graphid': u'7812', u'name': u'Network traffic on eth0'}
{u'graphid': u'8209', u'name': u'Network traffic on eth0'}
{u'graphid': u'8281', u'name': u'Network traffic on eth0'}
x is 0
y is 0
x is 1
y is 0
x is 0
y is 1
x is 1
y is 1
x is 0
y is 2
x is 1
y is 2
x is 0
y is 3
x is 1
y is 3
x is 0
y is 4
x is 1
y is 4
x is 0
y is 5
x is 1
y is 5
[{u'screenid': u'28', u'vsize': u'6', u'name': u'test Netowrk eth0', u'hsize': u'2', u'templateid': u'0'}]
The Given Screen Name Already Exists
28
Updating Screen Name test Netowrk eth0  Screen ID 28　　现在就是zabbix上查看数据即可：

　　个人使用了一下确实很方便！！希望可以帮助到各位！
　　
　　

　　

　　

　　

　　参考地址：
　　http://blog.运维网.com/john88wang