python批量添加zabbix Screens的两个脚本分享

前言

在最初搭建公司监控系统的时候，最头疼的是需要把同类项目组的相同图形添加到一个Screens,由于只能一个一个的添加，非常耗时耗经历。

下面分享两个脚本来解决这个头疼的问题。

1.将单个主机的所有图形添加到一个Screens

使用方法

    #更改main()函数里的url、username、password

    #参数一：主机名

    #参数二：筛选图名称

    python zabbix_screen_host.py 'zabbixserver' 'zabbixserver'

zabbix_screen_host.py脚本内容

    #!/usr/bin/env python
    #zabbix_screen_host.py
    import urllib2
    import json
    import argparse
    def authenticate(url, username, password):
     values = {'jsonrpc': '2.0',
     'method': 'user.login',
     'params': {
      'user': username,
      'password': password
     },
     'id': '0'
     }
     data = json.dumps(values)
     req = urllib2.Request(url, data, {'Content-Type': 'application/json-rpc'})
     response = urllib2.urlopen(req, data)
     output = json.loads(response.read())
     try:
     message = output['result']
     except:
     message = output['error']['data']
     print message
     quit()
     return output['result']
    def getGraph(hostname, url, auth, graphtype, dynamic, columns):
     if (graphtype == 0):
     selecttype = ['graphid']
     select = 'selectGraphs'
     if (graphtype == 1):
     selecttype = ['itemid', 'value_type']
     select = 'selectItems'
     values = {'jsonrpc': '2.0',
     'method': 'host.get',
     'params': {
      select: selecttype,
      'output': ['hostid', 'host'],
      'searchByAny': 1,
      'filter': {
      'host': hostname
      }
     },
     'auth': auth,
     'id': '2'
     }
     data = json.dumps(values)
     req = urllib2.Request(url, data, {'Content-Type': 'application/json-rpc'})
     response = urllib2.urlopen(req, data)
     host_get = response.read()
     output = json.loads(host_get)
     # print json.dumps(output)
     graphs = []
     if (graphtype == 0):
     for i in output['result'][0]['graphs']:
     graphs.append(i['graphid'])
     if (graphtype == 1):
     for i in output['result'][0]['items']:
     if int(i['value_type']) in (0, 3):
     graphs.append(i['itemid'])
     graph_list = []
     x = 0
     y = 0
     for graph in graphs:
     graph_list.append({
     "resourcetype": graphtype,
     "resourceid": graph,
     "width": "500",
     "height": "100",
     "x": str(x),
     "y": str(y),
     "colspan": "1",
     "rowspan": "1",
     "elements": "0",
     "valign": "0",
     "halign": "0",
     "style": "0",
     "url": "",
     "dynamic": str(dynamic)
     })
     x += 1
     if x == columns:
     x = 0
     y += 1
     return graph_list
    def screenCreate(url, auth, screen_name, graphids, columns):
     # print graphids
     if len(graphids) % columns == 0:
     vsize = len(graphids) / columns
     else:
     vsize = (len(graphids) / columns) + 1
     values = {"jsonrpc": "2.0",
     "method": "screen.create",
     "params": [{
      "name": screen_name,
      "hsize": columns,
      "vsize": vsize,
      "screenitems": []
     }],
     "auth": auth,
     "id": 2
     }
     for i in graphids:
     values['params'][0]['screenitems'].append(i)
     data = json.dumps(values)
     req = urllib2.Request(url, data, {'Content-Type': 'application/json-rpc'})
     response = urllib2.urlopen(req, data)
     host_get = response.read()
     output = json.loads(host_get)
     try:
     message = output['result']
     except:
     message = output['error']['data']
     print json.dumps(message)

    def main():
     url = 'http://zabbixip/zabbix/api_jsonrpc.php'
     username = "***"
     password = "***"
     parser = argparse.ArgumentParser(description='Create Zabbix screen from all of a host Items or Graphs.')
     parser.add_argument('hostname', metavar='H', type=str,
      help='Zabbix Host to create screen from')
     parser.add_argument('screenname', metavar='N', 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, default=3,
      help='number of columns in the screen (default: 3)')
     parser.add_argument('-d', dest='dynamic', action='store_true',
      help='enable for dynamic screen items (default: disabled)')
     parser.add_argument('-t', dest='screentype', action='store_true',
      help='set to 1 if you want item simple graphs created (default: 0, regular graphs)')
     args = parser.parse_args()
     hostname = args.hostname
     screen_name = args.screenname
     columns = args.columns
     dynamic = (1 if args.dynamic else 0)
     screentype = (1 if args.screentype else 0)
     auth = authenticate(url, username, password)
     graphids = getGraph(hostname, url, auth, screentype, dynamic, columns)
     print "Screen Name: " + screen_name
     print "Total Number of Graphs: " + str(len(graphids))
     screenCreate(url, auth, screen_name, graphids, columns)
    if __name__ == '__main__':
     main()

2.将同组主机的同一图形添加到一个Screens

使用方法

    #更改main()函数里的url、username、password

    #-g ：组名称

    #-G：图形名称

    #-n ：筛选（screen）图名称

    #-c : 一行有多少图形

    python zabbix_screen_group.py -g 'zabbix' -G 'icmp-ping' -n 'zabbix-icmp-ping' -c 2

zabbix_screen_group.py脚本内容

    #!/usr/bin/env python
    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": "100",
      "x": str(x),
      "y": str(y),
      "colspan": "1",
      "rowspan": "1",
      "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://zabbixip/zabbix/api_jsonrpc.php'
     username = '****'
     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()

总结

以上就是这篇文章的全部内容了，希望本文的内容的内容对大家的学习或者工作能带来一定的帮助，如果有疑问大家可以留言交流。